fix(ingest/dbt): fix bug in CLL pruning (datahub-project#11614)

aviv-julienjehannet · Oct 21, 2024 · bb4a50a · bb4a50a
1 parent 2101ff1
commit bb4a50a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/dbt/dbt_common.py b/metadata-ingestion/src/datahub/ingestion/source/dbt/dbt_common.py
@@ -1074,8 +1074,8 @@ def add_node_to_cll_list(dbt_name: str) -> None:
             for upstream in all_nodes_map[dbt_name].upstream_nodes:
                 schema_nodes.add(upstream)
 
-                upstream_node = all_nodes_map[upstream]
-                if upstream_node.is_ephemeral_model():
+                upstream_node = all_nodes_map.get(upstream)
+                if upstream_node and upstream_node.is_ephemeral_model():
                     add_node_to_cll_list(upstream)
 
             cll_nodes.add(dbt_name)