fix: missing type for default values (#12)

Signed-off-by: Michele Dolfi <[email protected]>
DS4SD · Jul 17, 2024 · d1d1724 · d1d1724
1 parent 2baa35c
commit d1d1724
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docling/datamodel/document.py b/docling/datamodel/document.py
@@ -117,9 +117,9 @@ class ConvertedDocument(BaseModel):
     errors: List[Dict] = []  # structure to keep errors
 
     pages: List[Page] = []
-    assembled: AssembledUnit = None
+    assembled: Optional[AssembledUnit] = None
 
-    output: DsDocument = None
+    output: Optional[DsDocument] = None
 
     def to_ds_document(self) -> DsDocument:
         title = ""