fixes and cleanup

marker/v2/processors/sectionheader.py

@@ -26,7 +26,7 @@ def __call__(self, document: Document):
                 if block.block_type not in self.block_types:
                     continue
 
-                line_heights[block.block_id] = [document.get_block(l).polygon.height for l in block.structure if l.block_type == BlockTypes.Line]
+                line_heights[block.block_id] = [document.get_block(l).polygon.height for l in block.structure if block.structure is not None and l.block_type == BlockTypes.Line]
 
         flat_line_heights = [h for heights in line_heights.values() for h in heights]
         heading_ranges = self.bucket_headings(flat_line_heights)

marker/v2/schema/blocks/inlinemath.py

@@ -10,7 +10,7 @@ def assemble_html(self, child_blocks, parent_structure):
         template = super().assemble_html(child_blocks, parent_structure)
         template = template.replace("\n", " ")
 
-        class_name = "text"
+        class_attr = ""
         if self.has_continuation:
-            class_name += " _has-continuation"
-        return f"<p class='{class_name}'>{template}</p>"
+            class_attr = " class='has-continuation'"
+        return f"<p {class_attr}>{template}</p>"

marker/v2/schema/blocks/text.py

@@ -13,4 +13,4 @@ def assemble_html(self, child_blocks, parent_structure):
         class_attr = ""
         if self.has_continuation:
             class_attr += "class='has-continuation'"
-        return f"<p {class_attr}'>{template}</p>"
+        return f"<p {class_attr}>{template}</p>"