Added pyarrow in the list of packages to install

paypal · Mar 5, 2024 · 9bf4494 · 9bf4494
1 parent b6275a2
commit 9bf4494
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -86,10 +86,11 @@
     install_requires=[
         "scikit-learn>=1.0.1",
         "pandas",
+        "pyarrow",
         "lightgbm",
         "xgboost",
-        "treelite",
-        "treelite-runtime",
+        # "treelite",
+        # "treelite-runtime",
         "dill",
     ],
     extras_require={
@@ -110,7 +111,6 @@
             "black",
             "isort",
             "pyspark",
-            "pyspark",
             "distributed",
             "dask",
             "numpy",