[CI] Fix h5py dependency in olddeps

ghstack-source-id: 87936dc1f6a55f89e9a1b2fe99e5c2cd12ae579b Pull Request resolved: pytorch#2513
kurtamohler · Oct 22, 2024 · 5244a90 · 5244a90
1 parent 1431ae5
commit 5244a90
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/.github/unittest/linux_libs/scripts_rlhf/install.sh b/.github/unittest/linux_libs/scripts_rlhf/install.sh
@@ -31,15 +31,15 @@ git submodule sync && git submodule update --init --recursive
 printf "Installing PyTorch with cu121"
 if [[ "$TORCH_VERSION" == "nightly" ]]; then
   if [ "${CU_VERSION:-}" == cpu ] ; then
-      pip3 install --pre torch numpy==1.26.4 --index-url https://download.pytorch.org/whl/nightly/cpu -U
+      pip3 install --pre torch "numpy<2.0.0" --index-url https://download.pytorch.org/whl/nightly/cpu -U
   else
-      pip3 install --pre torch numpy==1.26.4 --index-url https://download.pytorch.org/whl/nightly/cu121 -U
+      pip3 install --pre torch "numpy<2.0.0" --index-url https://download.pytorch.org/whl/nightly/cu121 -U
   fi
 elif [[ "$TORCH_VERSION" == "stable" ]]; then
     if [ "${CU_VERSION:-}" == cpu ] ; then
-      pip3 install torch numpy==1.26.4 --index-url https://download.pytorch.org/whl/cpu
+      pip3 install torch "numpy<2.0.0" --index-url https://download.pytorch.org/whl/cpu
   else
-      pip3 install torch numpy==1.26.4 --index-url https://download.pytorch.org/whl/cu121
+      pip3 install torch "numpy<2.0.0" --index-url https://download.pytorch.org/whl/cu121
   fi
 else
   printf "Failed to install pytorch"

diff --git a/.github/unittest/linux_olddeps/scripts_gym_0_13/environment.yml b/.github/unittest/linux_olddeps/scripts_gym_0_13/environment.yml
@@ -27,3 +27,4 @@ dependencies:
     - pyopengl==3.1.4
     - ray
     - av
+    - h5py
diff --git a/test/test_cost.py b/test/test_cost.py
@@ -157,6 +157,9 @@
     pytest.mark.filterwarnings(
         "ignore:The current behavior of MLP when not providing `num_cells` is that the number"
     ),
+    pytest.mark.filterwarnings(
+        "ignore:dep_util is Deprecated. Use functions from setuptools instead"
+    ),
 ]
-Original file line number
+Diff line change
@@ Expand Up / @@ -27,3 +27,4 @@ dependencies: @@
         - pyopengl==3.1.4
         - ray
         - av
+        - h5py