Fix return type annotation (#16772)

This commit is contained in:
Ayush Shah 2024-06-24 19:27:59 +05:30 committed by GitHub
parent b05d29ccee
commit e9c7c398c5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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.