Skip to content

Commit

Permalink
fix typing
Browse files Browse the repository at this point in the history
  • Loading branch information
AstrakhantsevaAA committed Nov 8, 2023
1 parent 9886f5f commit 9ef7a7e
Showing 1 changed file with 5 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Expand Down

0 comments on commit 9ef7a7e

Please sign in to comment.