Use numpy<2 for unit tests with torch<=2.2

fferflo · Nov 29, 2024 · 22766b8 · 22766b8
1 parent 994643d
commit 22766b8
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/.github/workflows/run_pytest.yml b/.github/workflows/run_pytest.yml
@@ -64,11 +64,11 @@ jobs:
         EINX_FILTER_TRACEBACK=0 pytest test --backend torch
         pip install --upgrade "torch==2.3.0"
         EINX_FILTER_TRACEBACK=0 pytest test --backend torch
-        pip install --upgrade "torch==2.2.0"
+        pip install --upgrade "torch==2.2.0" numpy<2
         EINX_FILTER_TRACEBACK=0 pytest test --backend torch
-        pip install --upgrade "torch==2.1.0"
+        pip install --upgrade "torch==2.1.0" numpy<2
         EINX_FILTER_TRACEBACK=0 pytest test --backend torch
-        pip install --upgrade "torch==2.0.0"
+        pip install --upgrade "torch==2.0.0" numpy<2
         EINX_FILTER_TRACEBACK=0 pytest test --backend torch
-        pip install --upgrade "torch<2"
+        pip install --upgrade "torch<2" numpy<2
         EINX_FILTER_TRACEBACK=0 pytest test/test_invalid_backend_version.py