Fix column lineage nonetype error (#13432)

This commit is contained in:
Mayur Singal 2023-10-04 17:48:56 +05:30 committed by GitHub
parent 2e49b72ccf
commit 3b640b43b7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -252,6 +252,7 @@ def _build_table_lineage(
""" """
Prepare the lineage request generator Prepare the lineage request generator
""" """
if from_entity and to_entity:
col_lineage = get_column_lineage( col_lineage = get_column_lineage(
to_entity=to_entity, to_entity=to_entity,
to_table_raw_name=str(to_table_raw_name), to_table_raw_name=str(to_table_raw_name),
@ -262,7 +263,6 @@ def _build_table_lineage(
lineage_details = LineageDetails(sqlQuery=query, source=lineage_source) lineage_details = LineageDetails(sqlQuery=query, source=lineage_source)
if col_lineage: if col_lineage:
lineage_details.columnsLineage = col_lineage lineage_details.columnsLineage = col_lineage
if from_entity and to_entity:
lineage = AddLineageRequest( lineage = AddLineageRequest(
edge=EntitiesEdge( edge=EntitiesEdge(
fromEntity=EntityReference( fromEntity=EntityReference(