Update

[ghstack-poisoned]
pytorch · Oct 22, 2024 · 9be6bdf · 9be6bdf
1 parent 2f45ad9
commit 9be6bdf
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==1.26.3 --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==1.26.3 --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==1.26.3 --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==1.26.3 --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"
+    ),
 ]