Skip to content

Commit

Permalink
[CI] Fix h5py dependency in olddeps
Browse files Browse the repository at this point in the history
ghstack-source-id: c81cf42597ba3ce9470986f49b32ead675c10fbe
Pull Request resolved: #2513
  • Loading branch information
vmoens committed Oct 22, 2024
1 parent 1431ae5 commit dccd1f1
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 4 deletions.
8 changes: 4 additions & 4 deletions .github/unittest/linux_libs/scripts_rlhf/install.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -27,3 +27,4 @@ dependencies:
- pyopengl==3.1.4
- ray
- av
- h5py
3 changes: 3 additions & 0 deletions test/test_cost.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"
),
]


Expand Down

0 comments on commit dccd1f1

Please sign in to comment.