From 9bf4494c49abe1f6a215a78f4a0dfee3ba219362 Mon Sep 17 00:00:00 2001
From: Charles <cpoli374@gmail.com>
Date: Tue, 5 Mar 2024 10:58:11 +0000
Subject: [PATCH] Added pyarrow in the list of packages to install

---
 setup.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index b9a62b1..2c52a5b 100644
--- 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",