refactor(lineage): simplify Node.walk() (#3098)

tobymao · Mar 8, 2024 · cea7508 · cea7508
1 parent 3bbcca2
commit cea7508
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/sqlglot/lineage.py b/sqlglot/lineage.py
@@ -28,10 +28,7 @@ def walk(self) -> t.Iterator[Node]:
         yield self
 
         for d in self.downstream:
-            if isinstance(d, Node):
-                yield from d.walk()
-            else:
-                yield d
+            yield from d.walk()
 
     def to_html(self, dialect: DialectType = None, **opts) -> GraphHTML:
         nodes = {}