fix(ingest/unity): Add sqllineage dependency (datahub-project#7938)

tejbloke · May 2, 2023 · 4e9c398 · 4e9c398
1 parent 5c1283a
commit 4e9c398
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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.