diff --git a/ingestion/src/metadata/ingestion/source/dashboard/powerbi/metadata.py b/ingestion/src/metadata/ingestion/source/dashboard/powerbi/metadata.py index f45e9f0db473..685e699c281d 100644 --- a/ingestion/src/metadata/ingestion/source/dashboard/powerbi/metadata.py +++ b/ingestion/src/metadata/ingestion/source/dashboard/powerbi/metadata.py @@ -544,12 +544,12 @@ def create_datamodel_report_lineage( to_entity=report_entity, from_entity=datamodel_entity ) - # create the lineage between table and datamodel - yield from self.create_table_datamodel_lineage( - db_service_name=db_service_name, - tables=dataset.tables, - datamodel_entity=datamodel_entity, - ) + for table in dataset.tables or []: + yield self._get_table_and_datamodel_lineage( + db_service_name=db_service_name, + table=table, + datamodel_entity=datamodel_entity, + ) # create the lineage between table and datamodel using the pbit files if self.client.file_client: @@ -678,20 +678,6 @@ def create_table_datamodel_lineage_from_files( ) ) - def create_table_datamodel_lineage( - self, - db_service_name: str, - tables: Optional[List[PowerBiTable]], - datamodel_entity: Optional[DashboardDataModel], - ) -> Iterable[Either[CreateDashboardRequest]]: - """Method to create lineage between table and datamodels""" - for table in tables or []: - yield self._get_table_and_datamodel_lineage( - db_service_name=db_service_name, - table=table, - datamodel_entity=datamodel_entity, - ) - def yield_dashboard_lineage_details( self, dashboard_details: Union[PowerBIDashboard, PowerBIReport], diff --git a/openmetadata-ui/src/main/resources/ui/src/constants/Lineage.constants.ts b/openmetadata-ui/src/main/resources/ui/src/constants/Lineage.constants.ts index 985bfe26cee3..3bf79b3fa7cc 100644 --- a/openmetadata-ui/src/main/resources/ui/src/constants/Lineage.constants.ts +++ b/openmetadata-ui/src/main/resources/ui/src/constants/Lineage.constants.ts @@ -30,6 +30,7 @@ export const DATATYPES_HAVING_SUBFIELDS = [ 'STRUCT', 'ARRAY', 'UNION', + 'TABLE', ]; export const PIPELINE_EDGE_WIDTH = 200;