From 7bbc4e35b3b7a102d503be94f7838c8e3d806356 Mon Sep 17 00:00:00 2001 From: sagar-salvi-apptware Date: Fri, 29 Nov 2024 19:32:49 +0530 Subject: [PATCH] fix(ingestion/dremio): Fixed lineage view for dremio EE --- .../src/datahub/ingestion/source/dremio/dremio_source.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_source.py b/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_source.py index 5b96845ec0496..b40e1398d2913 100644 --- a/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_source.py +++ b/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_source.py @@ -394,10 +394,12 @@ def process_dataset( ): yield dremio_mcp # Check if the emitted aspect is SchemaMetadataClass - if isinstance(dremio_mcp.metadata, SchemaMetadataClass): + if isinstance( + dremio_mcp.metadata, MetadataChangeProposalWrapper + ) and isinstance(dremio_mcp.metadata.aspect, SchemaMetadataClass): self.sql_parsing_aggregator.register_schema( urn=dataset_urn, - schema=dremio_mcp.metadata, + schema=dremio_mcp.metadata.aspect, ) if dataset_info.dataset_type == DremioDatasetType.VIEW: