fix typing

dlt-hub · Nov 8, 2023 · 9886f5f · 9886f5f
1 parent 6b65e4c
commit 9886f5f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docs/examples/nested_data/nested_data.py b/docs/examples/nested_data/nested_data.py
@@ -27,13 +27,13 @@ def mongodb_collection(
     write_disposition: Optional[str] = dlt.config.value,
 ) -> Any:
     # set up mongo client
-    client = MongoClient(connection_url, uuidRepresentation="standard", tz_aware=True)
+    client: Any = MongoClient(connection_url, uuidRepresentation="standard", tz_aware=True)
     mongo_database = client.get_default_database() if not database else client[database]
     collection_obj = mongo_database[collection]
 
     def collection_documents(
         client: Any,
-        collection: str,
+        collection: Any,
         incremental: Optional[dlt.sources.incremental[Any]] = None,
     ) -> Iterator[TDataItem]:
         LoaderClass = CollectionLoader
@@ -53,7 +53,7 @@ class CollectionLoader:
     def __init__(
         self,
         client: Any,
-        collection: str,
+        collection: Any,
         incremental: Optional[dlt.sources.incremental[Any]] = None,
     ) -> None:
         self.client = client