Skip to content

Commit

Permalink
fix keyerror
Browse files Browse the repository at this point in the history
  • Loading branch information
hsheth2 committed Oct 13, 2023
1 parent e70d463 commit 7b3147f
Showing 1 changed file with 4 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -855,6 +855,10 @@ def _infer_schemas_and_update_cll(self, all_nodes_map: Dict[str, DBTNode]) -> No
)
for column_lineage_info in sql_result.column_lineage
for upstream_column in column_lineage_info.upstreams
# Only include the CLL if the table in in the upstream list.
if target_platform_urn_to_dbt_name.get(upstream_column.table)
and target_platform_urn_to_dbt_name[upstream_column.table]
in node.upstream_nodes
]

# If we didn't fetch the schema from the graph, use the inferred schema.
Expand Down

0 comments on commit 7b3147f

Please sign in to comment.