diff --git a/docs/website/docs/examples/nested_data/code/nested_data-snippets.py b/docs/website/docs/examples/nested_data/code/nested_data-snippets.py index cda248a4fc..d6328fc2c5 100644 --- a/docs/website/docs/examples/nested_data/code/nested_data-snippets.py +++ b/docs/website/docs/examples/nested_data/code/nested_data-snippets.py @@ -30,13 +30,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, - collection, + client: Any, + collection: Any, incremental: Optional[dlt.sources.incremental[Any]] = None, ) -> Iterator[TDataItem]: LoaderClass = CollectionLoader @@ -56,8 +56,8 @@ def collection_documents( class CollectionLoader: def __init__( self, - client, - collection, + client: Any, + collection: Any, incremental: Optional[dlt.sources.incremental[Any]] = None, ) -> None: self.client = client