Rewrite installation botorch and pytorch

optuna · Sep 10, 2022 · 2e171da · 2e171da
1 parent 7be30fd
commit 2e171da
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -67,15 +67,15 @@ jobs:
         pip install cma
         pip install shap
 
-    # TODO(nzw): Move PyTorch Installation to the end of `Install` after Optuna drops Python 3.6 support.
-    - name: Install PyTorch and BoTorch with Python>3.7.
-      run: pip install botorch torch==1.11.0 --extra-index-url https://download.pytorch.org/whl/cpu
-      if: matrix.python-version > '3.7'
+        if [ ${{ matrix.python-version }} != 3.6 ]; then
+          pip install torch==1.11.0 --extra-index-url https://download.pytorch.org/whl/cpu
 
-    # TODO(nzw): Remove GPyTorch when Optuna drops BoTorch support with Python 3.7.
-    - name: Install PyTorch and BoTorch  with Python==3.7.
-      run: pip install botorch "gpytorch<1.9.0" torch==1.11.0 --extra-index-url https://download.pytorch.org/whl/cpu
-      if: matrix.python-version == '3.7'
+          if [ ${{ matrix.python-version }} == 3.7 ]; then
+            pip install botorch "gpytorch<1.9.0"
+          elif [ ${{ matrix.python-version }} != 3.7 ]; then
+            pip install botorch
+          fi
+        fi
 
     - name: Tests
       run: |