Skip to content

Commit 0616961

Browse files
committed
fixed table logic
1 parent 6306c9a commit 0616961

File tree

1 file changed

+7
-3
lines changed
  • metadata-ingestion/src/datahub/ingestion/source

1 file changed

+7
-3
lines changed

metadata-ingestion/src/datahub/ingestion/source/superset.py

+7-3
Original file line numberDiff line numberDiff line change
@@ -750,7 +750,7 @@ def generate_physical_dataset_lineage(
750750
cll.append(
751751
ColumnLineageInfo(
752752
downstream=DownstreamColumnRef(
753-
table=None,
753+
table=datasource_urn,
754754
column=column.get("column_name", ""),
755755
native_column_type=column.get("type", ""),
756756
),
@@ -768,8 +768,12 @@ def generate_physical_dataset_lineage(
768768

769769
for cll_info in cll:
770770
downstream = (
771-
[make_schema_field_urn(datasource_urn, cll_info.downstream.column)]
772-
if cll_info.downstream and cll_info.downstream.column
771+
[
772+
make_schema_field_urn(
773+
cll_info.downstream.table, cll_info.downstream.column
774+
)
775+
]
776+
if cll_info.downstream
773777
else []
774778
)
775779
upstreams = [

0 commit comments

Comments
 (0)