fix depth

ttusing · Aug 9, 2024 · 9b8e887 · 9b8e887
1 parent f4eeff9
commit 9b8e887
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/dbt/graph/graph.py b/core/dbt/graph/graph.py
@@ -67,7 +67,7 @@ def select_children(
         while len(selected) > 0 and (max_depth is None or i < max_depth):
             next_layer: Set[UniqueId] = set()
             for node in selected:
-                next_layer.update(self.graph.descendants(node))
+                next_layer.update(self.descendants(node, 1))
             next_layer = next_layer - children  # Avoid re-searching
             children.update(next_layer)
             selected = next_layer
@@ -86,7 +86,7 @@ def select_parents(
         while len(selected) > 0 and (max_depth is None or i < max_depth):
             next_layer: Set[UniqueId] = set()
             for node in selected:
-                next_layer.update(self.graph.ancestors(node))
+                next_layer.update(self.ancestors(node, 1))
             next_layer = next_layer - parents  # Avoid re-searching
             parents.update(next_layer)
             selected = next_layer