diff --git a/.github/unittest/linux_libs/scripts_rlhf/install.sh b/.github/unittest/linux_libs/scripts_rlhf/install.sh
index 9a5cf82074b..2ec0b17b2bc 100755
--- 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
index 06c4a112933..042a447a1e1 100644
--- 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
index f6b0ed21936..fc21512da74 100644
--- 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"
+    ),
 ]