Minor fixes

Author: VikParuchuri

marker/converters/pdf.py

@@ -1,5 +1,7 @@
 import os
 
+from marker.schema.document import Document
+
 os.environ["TOKENIZERS_PARALLELISM"] = "false"  # disables a tokenizers warning
 
 from collections import defaultdict
@@ -171,7 +173,7 @@ def filepath_to_str(self, file_input: Union[str, io.BytesIO]):
             if temp_file is not None and os.path.exists(temp_file.name):
                 os.unlink(temp_file.name)
 
-    def build_document(self, filepath: str):
+    def build_document(self, filepath: str) -> Document:
         provider_cls = provider_from_filepath(filepath)
         layout_builder = self.resolve_dependencies(self.layout_builder_class)
         line_builder = self.resolve_dependencies(LineBuilder)

marker/schema/blocks/base.py

@@ -115,6 +115,16 @@ def from_block(cls, block: Block) -> Block:
         block_attrs = block.model_dump(exclude=["id", "block_id", "block_type"])
         return cls(**block_attrs)
 
+    def set_internal_metadata(self, key, data):
+        if self._metadata is None:
+            self._metadata = {}
+        self._metadata[key] = data
+
+    def get_internal_metadata(self, key):
+        if self._metadata is None:
+            return None
+        return self._metadata.get(key)
+
     def get_image(
         self,
         document: Document,