pin numpy < 2.0.0 and pyarrow < 15.0.0

pang-wu · Jun 16, 2024 · 4db9865 · 4db9865
1 parent b290119
commit 4db9865
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/setup.py b/python/setup.py
@@ -95,12 +95,12 @@ def run(self):
     copy2(SCRIPT_PATH, SCRIPT_TARGET)
 
     install_requires = [
-        "numpy",
+        "numpy < 2.0.0",
         "pandas >= 1.1.4",
         "psutil",
-        "pyarrow >= 4.0.1",
+        "pyarrow >= 4.0.1, <15.0.0",
         "ray >= 2.1.0",
-        "pyspark >= 3.1.1, <= 3.5.0",
+        "pyspark >= 3.1.1, <=3.5.1",
         "netifaces",
         "protobuf > 3.19.5, <= 3.20.3"
     ]