Remove the version constraint of GPyTorch

optuna · Sep 12, 2022 · be8f229 · be8f229
1 parent 79f0329
commit be8f229
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -69,8 +69,7 @@ jobs:
 
     # TODO(nzw): Move `run` part to the end of `Install` after Optuna drops Python 3.6 support.
     - name: Install PyTorch and BoTorch.
-      # TODO(nzw0301): Remove gpytorch if botorch supports gpytorch>1.8.
-      run: pip install botorch "gpytorch<1.9.0" torch==1.11.0 --extra-index-url https://download.pytorch.org/whl/cpu
+      run: pip install botorch torch==1.11.0 --extra-index-url https://download.pytorch.org/whl/cpu
       if: matrix.python-version != '3.6'
 
     - name: Tests

diff --git a/setup.py b/setup.py
@@ -52,8 +52,6 @@ def get_extras_require() -> Dict[str, List[str]]:
             "asv>=0.5.0",
             "botorch",
             "cma",
-            # TODO(nzw0301): Remove gpytorch if botorch supports gpytorch>1.8.
-            "gpytorch<1.9.0",
             "scikit-optimize",
             "virtualenv",
         ],
@@ -102,8 +100,6 @@ def get_extras_require() -> Dict[str, List[str]]:
             "chainer>=5.0.0",
             "cma",
             "fastai ; python_version>'3.6'",
-            # TODO(nzw0301): Remove gpytorch if botorch supports gpytorch>1.8.
-            "gpytorch<1.9.0 ; python_version>'3.6'",
             "lightgbm",
             "mlflow",
             "mpi4py",