diff options
Diffstat (limited to 'sqlglot/lineage.py')
| -rw-r--r-- | sqlglot/lineage.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/sqlglot/lineage.py b/sqlglot/lineage.py index a39ad8c..908f126 100644 --- a/sqlglot/lineage.py +++ b/sqlglot/lineage.py @@ -6,6 +6,7 @@ from dataclasses import dataclass, field from sqlglot import Schema, exp, maybe_parse from sqlglot.optimizer import Scope, build_scope, optimize +from sqlglot.optimizer.expand_laterals import expand_laterals from sqlglot.optimizer.qualify_columns import qualify_columns from sqlglot.optimizer.qualify_tables import qualify_tables @@ -38,7 +39,7 @@ def lineage( sql: str | exp.Expression, schema: t.Optional[t.Dict | Schema] = None, sources: t.Optional[t.Dict[str, str | exp.Subqueryable]] = None, - rules: t.Sequence[t.Callable] = (qualify_tables, qualify_columns), + rules: t.Sequence[t.Callable] = (qualify_tables, qualify_columns, expand_laterals), dialect: DialectType = None, ) -> Node: """Build the lineage graph for a column of a SQL query. |