Skip to content

Commit

Permalink
Hack around missing pytorch 1.13.1 binaries for py3.11 on mac (#1966)
Browse files Browse the repository at this point in the history
Summary:
This fix is needed to avoid errors in the CI

Pull Request resolved: #1966

Reviewed By: saitcakmak

Differential Revision: D47978234

Pulled By: Balandat

fbshipit-source-id: c5024a36ce8a0952debdea4aa1cbc3ffb925031d
  • Loading branch information
Balandat authored and facebook-github-bot committed Aug 2, 2023
1 parent 3b674eb commit 85cec4c
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions .github/workflows/test_stable.yml
Original file line number Diff line number Diff line change
Expand Up @@ -70,10 +70,15 @@ jobs:
with:
python-version: ${{ matrix.python-version }}
- name: Install dependencies
env:
PYTHON_VERSION: ${{ matrix.python-version }}
OS: ${{ matrix.os }}
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" ]] && [[ "${OS}"=="macos-latest" ]]; 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 85cec4c

Please sign in to comment.