fix: limit numpy version to < 2.0 and fix missing dependency (#10537)

Oneflow-Inc · Jun 20, 2024 · 850b4ad · 850b4ad
1 parent 9cf3ee5
commit 850b4ad
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/dev-requirements.txt b/dev-requirements.txt
@@ -1,6 +1,6 @@
 black==19.10b0; python_version >= "3.6"
 click==8.0.0; python_version >= "3.6" # https://github.com/psf/black/issues/2964
-numpy>=1.21.6
+numpy>=1.21.6, <2.0
 protobuf>=3.9.2, <4.0
 wheel
 tqdm
@@ -16,3 +16,4 @@ pytest-xdist
 pytest-repeat
 rich
 portalocker
+typing-extensions>=4.0.0, <5.0
diff --git a/python/oneflow/mock_torch/mock_utils.py b/python/oneflow/mock_torch/mock_utils.py
@@ -19,7 +19,7 @@
 from collections import deque
 from importlib import import_module
 
-if sys.version_info < (3, 8):
+if sys.version_info <= (3, 8):
     try:
         from importlib_metadata import requires
     except ImportError:

diff --git a/python/setup.py b/python/setup.py
@@ -53,8 +53,9 @@ def get_version():
 
 
 REQUIRED_PACKAGES = [
-    f"numpy>={np.__version__}",
+    f"numpy>={np.__version__}, <2.0",
     "protobuf>=3.9.2, <4.0",
+    "typing-extensions>=4.0.0, <5.0",
     "tqdm",
     "requests",
     "pillow",