From 9be6bdffae00cc8df37c140cd8f73ca1ece5724d Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Tue, 22 Oct 2024 12:35:36 -0700 Subject: [PATCH 1/2] Update [ghstack-poisoned] --- .github/unittest/linux_libs/scripts_rlhf/install.sh | 8 ++++---- .../linux_olddeps/scripts_gym_0_13/environment.yml | 1 + test/test_cost.py | 3 +++ 3 files changed, 8 insertions(+), 4 deletions(-) diff --git a/.github/unittest/linux_libs/scripts_rlhf/install.sh b/.github/unittest/linux_libs/scripts_rlhf/install.sh index 9a5cf82074b..1bc77338ecd 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==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 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" + ), ] From d0287bee91ff6d298748728976cd6ac0e1c499a8 Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Tue, 22 Oct 2024 12:51:35 -0700 Subject: [PATCH 2/2] Update [ghstack-poisoned] --- .github/unittest/linux_libs/scripts_rlhf/install.sh | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/unittest/linux_libs/scripts_rlhf/install.sh b/.github/unittest/linux_libs/scripts_rlhf/install.sh index 1bc77338ecd..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.3 --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.3 --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.3 --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.3 --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"