From 4e9c398e1d43fc005d6684bf097864873bec0d83 Mon Sep 17 00:00:00 2001
From: Andrew Sikowitz <andrew.sikowitz@acryl.io>
Date: Mon, 1 May 2023 23:26:49 -0400
Subject: [PATCH] fix(ingest/unity): Add sqllineage dependency (#7938)

---
 metadata-ingestion/setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/metadata-ingestion/setup.py b/metadata-ingestion/setup.py
index c596030de5e47f..c26df811cb6572 100644
--- a/metadata-ingestion/setup.py
+++ b/metadata-ingestion/setup.py
@@ -240,7 +240,7 @@ def get_long_description():
     "sqlparse",
 }
 
-databricks_cli = {"databricks-cli==0.17.3", "pyspark"}
+databricks_cli = {"databricks-cli==0.17.3", "pyspark", "requests"}
 
 # Note: for all of these, framework_common will be added.
 plugins: Dict[str, Set[str]] = {
@@ -368,7 +368,7 @@ def get_long_description():
     "powerbi": microsoft_common | {"lark[regex]==1.1.4", "sqlparse"},
     "powerbi-report-server": powerbi_report_server,
     "vertica": sql_common | {"vertica-sqlalchemy-dialect[vertica-python]==0.0.1"},
-    "unity-catalog": databricks_cli | {"requests"} | usage_common,
+    "unity-catalog": databricks_cli | sqllineage_lib,
 }
 
 # This is mainly used to exclude plugins from the Docker image.