diff --git a/ingestion/setup.py b/ingestion/setup.py index 09775e366fe..2447b7b6b6b 100644 --- a/ingestion/setup.py +++ b/ingestion/setup.py @@ -118,7 +118,7 @@ base_requirements = { "requests-aws4auth~=1.1", # Only depends on requests as external package. Leaving as base. "setuptools~=66.0.0", "sqlalchemy>=1.4.0,<2", - "collate-sqllineage>=1.0.4", + "collate-sqllineage~=1.3.0", "tabulate==0.9.0", "typing-inspect", "wheel~=0.38.4", diff --git a/ingestion/src/metadata/ingestion/lineage/parser.py b/ingestion/src/metadata/ingestion/lineage/parser.py index 732afa85d6a..5d855784523 100644 --- a/ingestion/src/metadata/ingestion/lineage/parser.py +++ b/ingestion/src/metadata/ingestion/lineage/parser.py @@ -19,10 +19,10 @@ from typing import Any, Dict, List, Optional, Tuple import sqlparse from cached_property import cached_property -from sqllineage import SQLPARSE_DIALECT -from sqllineage.core.models import Column, Table -from sqllineage.exceptions import SQLLineageException -from sqllineage.runner import LineageRunner +from collate_sqllineage import SQLPARSE_DIALECT +from collate_sqllineage.core.models import Column, Table +from collate_sqllineage.exceptions import SQLLineageException +from collate_sqllineage.runner import LineageRunner from sqlparse.sql import Comparison, Identifier, Parenthesis, Statement from metadata.generated.schema.type.tableUsageCount import TableColumn, TableColumnJoin diff --git a/ingestion/tests/unit/test_query_parser.py b/ingestion/tests/unit/test_query_parser.py index 12fdc806b64..3903183b293 100644 --- a/ingestion/tests/unit/test_query_parser.py +++ b/ingestion/tests/unit/test_query_parser.py @@ -15,7 +15,7 @@ Validate query parser logic from unittest import TestCase -from sqllineage.core.models import Column +from collate_sqllineage.core.models import Column from metadata.generated.schema.type.tableUsageCount import TableColumn, TableColumnJoin from metadata.ingestion.lineage.models import Dialect