Fix mypy

aryn-ai · Feb 14, 2025 · e608ab5 · e608ab5
1 parent e8bd9d0
commit e608ab5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/sycamore/sycamore/connectors/aryn/ArynReader.py b/lib/sycamore/sycamore/connectors/aryn/ArynReader.py
@@ -152,9 +152,9 @@ def _to_doc(self, doc: dict[str, Any]) -> dict[str, Any]:
 
         doc = aryn_client.get_doc(self._query_params.docset_id, doc["doc_id"])
         elements = doc.get("elements", [])
-        doc = Document(**doc)
-        doc.data["elements"] = [create_element(**element) for element in elements]
-        return {"doc": Document.serialize(doc)}
+        document = Document(**doc)
+        document.data["elements"] = [create_element(**element) for element in elements]
+        return {"doc": Document.serialize(document)}
 
     def execute(self, **kwargs) -> "Dataset":