From 9886f5fca62fd65fd24bd456a9376de31abb1b42 Mon Sep 17 00:00:00 2001 From: AstrakhantsevaAA Date: Wed, 8 Nov 2023 12:12:21 +0100 Subject: [PATCH] fix typing --- docs/examples/nested_data/nested_data.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/examples/nested_data/nested_data.py b/docs/examples/nested_data/nested_data.py index b475b15c12..3e4a1295c3 100644 --- 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