Skip to content

Commit db22668

Browse files
authored
pin numpy for Python 3.8 (#7639)
1 parent 01b9faa commit db22668

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

setup.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,8 +57,10 @@ def write_version_file():
5757
if os.getenv("PYTORCH_VERSION"):
5858
pytorch_dep += "==" + os.getenv("PYTORCH_VERSION")
5959

60+
numpy_dep = "numpy" if sys.version_info[:2] >= (3, 9) else "numpy < 1.25"
61+
6062
requirements = [
61-
"numpy",
63+
numpy_dep,
6264
"requests",
6365
pytorch_dep,
6466
]

0 commit comments

Comments
 (0)