Skip to content

Commit

Permalink
Hack around the fact that pytorch does not have 1.13.1 binaries for m…
Browse files Browse the repository at this point in the history
…ac for py3.11 in CI
  • Loading branch information
Balandat committed Aug 1, 2023
1 parent 3b674eb commit 419f110
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions .github/workflows/test_stable.yml
Original file line number Diff line number Diff line change
Expand Up @@ -70,10 +70,14 @@ jobs:
with:
python-version: ${{ matrix.python-version }}
- name: Install dependencies
env:
PYTHON_VERSION: ${{ matrix.python-version }}
run: |
python setup.py egg_info
req_txt="botorch.egg-info/requires.txt"
min_torch_version=$(grep '\btorch[>=]=' ${req_txt} | sed 's/[^0-9.]//g')
# HACK around the fact that pytorch does not offer a mac binary for 1.13.1 for py3.11 - TODO: Remove when bumping torch to 2.0.1
min_torch_version=$(if [[ "$min_torch_version" == "1.13.1" ]] && [[ "$PYTHON_VERSION" >= "3.11" ]]; then echo "2.0.1"; else echo "$min_torch_version"; fi)
min_gpytorch_version=$(grep '\bgpytorch[>=]=' ${req_txt} | sed 's/[^0-9.]//g')
min_linear_operator_version=$(grep '\blinear_operator[>=]=' ${req_txt} | sed 's/[^0-9.]//g')
pip install "torch==${min_torch_version}" "gpytorch==${min_gpytorch_version}" "linear_operator==${min_linear_operator_version}"
Expand Down

0 comments on commit 419f110

Please sign in to comment.