From a9fc51ec8b8dbf6aac7c9004dc6101797aec6918 Mon Sep 17 00:00:00 2001
From: Mayur Singal <39544459+ulixius9@users.noreply.github.com>
Date: Mon, 5 Feb 2024 19:44:08 +0530
Subject: [PATCH] MINOR: Change sqllineage import to collate_sqllineage
 (#14870)

---
 ingestion/setup.py                                 | 2 +-
 ingestion/src/metadata/ingestion/lineage/parser.py | 8 ++++----
 ingestion/tests/unit/test_query_parser.py          | 2 +-
 3 files changed, 6 insertions(+), 6 deletions(-)

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