limit numpy version less than 2 (#2)

dnikolaev-amd · web-flow · commit 68ba7ec9a808 · 2024-07-08T13:39:33.000-05:00
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",
     pytorch_dep,
 ]
 

Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,7 @@ def write_version_file():`
`58`	`58`	`pytorch_dep += "==" + os.getenv("PYTORCH_VERSION")`
`59`	`59`
`60`	`60`	`requirements = [`
`61`		`- "numpy",`
	`61`	`+ "numpy<2",`
`62`	`62`	`pytorch_dep,`
`63`	`63`	`]`
`64`	`64`