pytorch
diff --git a/‎setup.py‎
Lines changed: 2 additions & 1 deletion b/‎setup.py‎
Lines changed: 2 additions & 1 deletion
@@ -58,7 +58,8 @@ def write_version_file():
  pytorch_dep += "==" + os.getenv("PYTORCH_VERSION")
 
 requirements = [
- "numpy",
+ # TODO: Remove <2 constraint! https://github.com/pytorch/vision/issues/8531
+ "numpy<2" if sys.platform == "win32" else "numpy",
  pytorch_dep,
 ]
Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,8 @@ def write_version_file():`
`58`	`58`	`pytorch_dep += "==" + os.getenv("PYTORCH_VERSION")`
`59`	`59`
`60`	`60`	`requirements = [`
`61`		`- "numpy",`
	`61`	`+ # TODO: Remove <2 constraint! https://github.com/pytorch/vision/issues/8531`
	`62`	`+ "numpy<2" if sys.platform == "win32" else "numpy",`
`62`	`63`	`pytorch_dep,`
`63`	`64`	`]`
`64`	`65`