From e9c7c398c5662a6b44c78d2c21c5f44e78e13261 Mon Sep 17 00:00:00 2001 From: Ayush Shah Date: Mon, 24 Jun 2024 19:27:59 +0530 Subject: [PATCH] Fix return type annotation (#16772) --- .../metadata/ingestion/source/database/datalake/metadata.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ingestion/src/metadata/ingestion/source/database/datalake/metadata.py b/ingestion/src/metadata/ingestion/source/database/datalake/metadata.py index 773e98334c2..1d60a896314 100644 --- a/ingestion/src/metadata/ingestion/source/database/datalake/metadata.py +++ b/ingestion/src/metadata/ingestion/source/database/datalake/metadata.py @@ -59,6 +59,7 @@ from metadata.ingestion.source.storage.storage_service import ( OPENMETADATA_TEMPLATE_FILE_NAME, ) from metadata.readers.dataframe.models import DatalakeTableSchemaWrapper +from metadata.readers.dataframe.reader_factory import SupportedTypes from metadata.readers.file.base import ReadException from metadata.readers.file.config_source_factory import get_reader from metadata.utils import fqn @@ -221,7 +222,7 @@ class DatalakeSource(DatabaseServiceSource): def get_tables_name_and_type( # pylint: disable=too-many-branches self, - ) -> Iterable[Tuple[str, TableType]]: + ) -> Iterable[Tuple[str, TableType, SupportedTypes]]: """ Handle table and views. @@ -262,7 +263,7 @@ class DatalakeSource(DatabaseServiceSource): yield table_name, TableType.Regular, file_extension def yield_table( - self, table_name_and_type: Tuple[str, TableType] + self, table_name_and_type: Tuple[str, TableType, SupportedTypes] ) -> Iterable[Either[CreateTableRequest]]: """ From topology.