Move SQL dependencies to extra

fugue-project · Jan 7, 2024 · be038de · be038de
1 parent 8008bfa
commit be038de
Showing 1 changed file with 13 additions and 17 deletions.
diff --git a/setup.py b/setup.py
@@ -4,6 +4,13 @@
 
 from fugue_version import __version__
 
+SQL_DEPENDENCIES = [
+    "qpd>=0.4.4",
+    "fugue-sql-antlr>=0.2.0",
+    "sqlglot",
+    "jinja2",
+]
+
 with open("README.md") as f:
     _text = ["# Fugue"] + f.read().splitlines()[1:]
     LONG_DESCRIPTION = "\n".join(_text)
@@ -33,19 +40,9 @@ def get_version() -> str:
     install_requires=[
         "triad>=0.9.3",
         "adagio>=0.2.4",
-        # sql dependencies
-        "qpd>=0.4.4",
-        "fugue-sql-antlr>=0.2.0",
-        "sqlglot",
-        "jinja2",
     ],
     extras_require={
-        "sql": [
-            "qpd>=0.4.4",
-            "fugue-sql-antlr>=0.2.0",
-            "sqlglot",
-            "jinja2",
-        ],
+        "sql": SQL_DEPENDENCIES,
         "cpp_sql_parser": ["fugue-sql-antlr[cpp]>=0.2.0"],
         "spark": ["pyspark>=3.1.1"],
         "dask": [
@@ -54,17 +51,16 @@ def get_version() -> str:
             "pandas>=2.0.2",
         ],
         "ray": ["ray[data]>=2.4.0", "duckdb>=0.5.0", "pyarrow>=6.0.1"],
-        "duckdb": [
+        "duckdb": SQL_DEPENDENCIES
+        + [
             "duckdb>=0.5.0",
             "numpy",
         ],
         "polars": ["polars"],
-        "ibis": ["ibis-framework>=3.2.0,<6"],
+        "ibis": SQL_DEPENDENCIES + ["ibis-framework>=3.2.0,<6"],
         "notebook": ["notebook", "jupyterlab", "ipython>=7.10.0"],
-        "all": [
-            "sqlglot",
-            "jinja2",
-            "fugue-sql-antlr[cpp]>=0.2.0",
+        "all": SQL_DEPENDENCIES
+        + [
             "pyspark>=3.1.1",
             "dask[distributed,dataframe]>=2023.5.0",
             "dask-sql",