fix(deps): Make deepgp optional install, torchvision dev install

.github/workflows/tests.yaml

@@ -37,7 +37,7 @@ jobs:
           poetry-version: 1.3.2
 
       - name: Run poetry install
-        run: poetry install --extras "gpytorch torchvision"
+        run: poetry install --extras "deepgp"
 
       - name: Run pytest
         timeout-minutes: 15

pyproject.toml

@@ -69,6 +69,7 @@ seaborn = "^0.13"
 pyyaml = "^6"
 tensorboard = "^2"
 # cython = "^3.0.4"
+gpytorch = { version="^1.8", optional=true}
 
 [tool.poetry.group.dev.dependencies]
 pre-commit = "^3"
@@ -80,13 +81,11 @@ typing-extensions = "^4"
 # jahs-bench = {git = "https://github.com/automl/jahs_bench_201.git", rev = "v1.0.2"}
 mkdocs-material = "^8.1.3"
 mike = "^1.1.2"
-gpytorch = { version="^1.8", optional=true}
-torchvision = { version="<0.16.0", optional=true}
+torchvision = "<0.16.0"  # Required by examples
 
 # https://python-poetry.org/docs/master/pyproject/#extras
 [tool.poetry.extras]
-gpytorch = ["gyptorch"]
-torchvision = ["torchvision"]
+deepgp = ["gyptorch"]
 
 [build-system]
 requires = ["poetry-core>=1.1.0"]

src/neps/optimizers/bayesian_optimization/models/__init__.py

@@ -6,7 +6,7 @@
 try:
     from .deepGP import DeepGP
 except ImportError as e:
-    DeepGP = MissingDependencyError(libname="neps", dep="gpytorch", install_group="gpytorch", cause=e)
+    DeepGP = MissingDependencyError(libname="neps", dep="gpytorch", install_group="deepgp", cause=e)
 
 try:
     from .pfn import PFN_SURROGATE  # only if available locally