From e5a721baa7d6d638603cab957d3e78a81e3ceb5d Mon Sep 17 00:00:00 2001 From: IlyaFaer Date: Mon, 15 Jul 2024 12:22:06 +0400 Subject: [PATCH] lint fix --- sources/mongodb/helpers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sources/mongodb/helpers.py b/sources/mongodb/helpers.py index c59e2702f..bf630db2c 100644 --- a/sources/mongodb/helpers.py +++ b/sources/mongodb/helpers.py @@ -164,7 +164,7 @@ def _create_batches(self, limit: Optional[int] = None) -> List[Dict[str, int]]: return batches - def _get_cursor(self, filter_: Dict[str, Any]) -> Cursor: + def _get_cursor(self, filter_: Dict[str, Any]) -> Cursor[Any]: """Get a reading cursor for the collection. Args: @@ -276,7 +276,7 @@ class CollectionArrowLoaderParallel(CollectionLoaderParallel): Apache Arrow for data processing. """ - def _get_cursor(self, filter_: Dict[str, Any]) -> Cursor: + def _get_cursor(self, filter_: Dict[str, Any]) -> Cursor[Any]: """Get a reading cursor for the collection. Args: