lint fix

dlt-hub · Jul 15, 2024 · e5a721b · e5a721b
1 parent 95a3210
commit e5a721b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: