limit numpy version less then 2 (#3)

(cherry picked from commit e041c21)
ROCm · Jul 10, 2024 · c5e28d6 · c5e28d6
1 parent c1ce9c9
commit c5e28d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -58,7 +58,7 @@ def write_version_file():
     pytorch_dep += "==" + os.getenv("PYTORCH_VERSION")
 
 requirements = [
-    "numpy",
+    "numpy<2",
     "requests",
     pytorch_dep,
 ]