Fixes for mypy

skrydal · Dec 18, 2024 · 9170064 · 9170064
1 parent 91b3902
commit 9170064
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/metadata-ingestion/tests/unit/test_iceberg.py b/metadata-ingestion/tests/unit/test_iceberg.py
@@ -65,7 +65,7 @@
 )
 
 
-def with_iceberg_source(processing_threads: int = 1, **kwargs) -> IcebergSource:
+def with_iceberg_source(processing_threads: int = 1, **kwargs: Any) -> IcebergSource:
     catalog = {"test": {"type": "rest"}}
     return IcebergSource(
         ctx=PipelineContext(run_id="iceberg-source-test"),
@@ -545,7 +545,7 @@ def __init__(self, tables: Dict[str, Dict[str, Callable[[], Table]]]):
         """
         self.tables = tables
 
-    def list_namespaces(self) -> Iterable[str]:
+    def list_namespaces(self) -> Iterable[Tuple[str]]:
         return [*[(key,) for key in self.tables.keys()]]
 
     def list_tables(self, namespace: str) -> Iterable[Tuple[str, str]]:
@@ -565,7 +565,7 @@ def list_tables(self, namespace: str) -> Iterable[Tuple[str, str]]:
 
 
 class MockCatalogExceptionListingNamespaces(MockCatalog):
-    def list_namespaces(self) -> Iterable[str]:
+    def list_namespaces(self) -> Iterable[Tuple[str]]:
         raise Exception()