Reduces the size of envs for CI

.github/workflows/python-tox-testing-ax.yml

@@ -17,7 +17,8 @@ jobs:
       max-parallel: 5
 
     steps:
-      - uses: actions/checkout@v3
+      - name: Checkout
+        uses: actions/checkout@v3
       - name: Set up Python 3.9
         uses: actions/setup-python@v3
         with:

src/poli_baselines/core/utils/ax/environment.ax.small.yml

@@ -0,0 +1,15 @@
+name: poli__ax
+channels:
+  - conda-forge
+  - defaults
+dependencies:
+  - python=3.10
+  - pytorch
+  - pip
+  - pip:
+    - scikit-learn
+    - botorch
+    - ax-platform
+    - numpy<2
+    - "git+https://github.com/MachineLearningLifeScience/[email protected]"
+    - "git+https://github.com/MachineLearningLifeScience/poli-baselines@main"

src/poli_baselines/solvers/bayesian_optimization/alebo/environment.alebo.small.yml

@@ -0,0 +1,15 @@
+name: poli__alebo
+channels:
+  - conda-forge
+  - defaults
+dependencies:
+  - python=3.10
+  - pytorch
+  - pip
+  - pip:
+    - scikit-learn
+    - botorch
+    - ax-platform==0.4.0
+    - numpy<2
+    - "git+https://github.com/MachineLearningLifeScience/[email protected]"
+    - "git+https://github.com/MachineLearningLifeScience/poli-baselines@main"

src/poli_baselines/solvers/bayesian_optimization/baxus/environment.baxus.small.yml

@@ -0,0 +1,13 @@
+name: poli__baxus
+channels:
+  - conda-forge
+  - defaults
+dependencies:
+  - python=3.10
+  - pytorch
+  - pip
+  - pip:
+    - numpy<2
+    - baxus
+    - "git+https://github.com/MachineLearningLifeScience/[email protected]"
+    - "git+https://github.com/MachineLearningLifeScience/poli-baselines@main"

src/poli_baselines/solvers/bayesian_optimization/pr/environment.pr.small.yml

@@ -0,0 +1,25 @@
+name: poli__pr
+channels:
+  - defaults
+dependencies:
+  - python=3.10
+  - pytorch
+  - pip
+  - pip:
+      - --extra-index-url https://download.pytorch.org/whl/cpu
+      - numpy<2
+      - torch
+      - gpytorch==1.11
+      - botorch==0.7.0
+      - scipy
+      - jupyter
+      - matplotlib
+      - nevergrad
+      - scikit-learn
+      - statsmodels
+      - xgboost
+      - black
+      - "git+https://github.com/miguelgondu/bo_pr.git@main"
+      - "git+https://github.com/MachineLearningLifeScience/[email protected]"
+      - "git+https://github.com/MachineLearningLifeScience/poli-baselines.git@main"
+

tox.ini

@@ -46,7 +46,7 @@ allowlist_externals =
     sh
     conda
 commands =
-    sh -c 'if conda info --envs | grep -q poli__baxus; then echo "poli__baxus already exists"; else conda env create -f ./src/poli_baselines/solvers/bayesian_optimization/baxus/environment.baxus.yml; fi'
+    sh -c 'if conda info --envs | grep -q poli__baxus; then echo "poli__baxus already exists"; else conda env create -f ./src/poli_baselines/solvers/bayesian_optimization/baxus/environment.baxus.small.yml; fi'
     sh -c "conda run -n poli__baxus python -m pip install -e ."
     sh -c "conda run -n poli__baxus python -m pip install pytest"
     conda run -n poli__baxus pytest {tty:--color=yes} -v src/poli_baselines/tests/solvers/bayesian_optimization/test_baxus.py {posargs}
@@ -59,7 +59,7 @@ allowlist_externals =
     sh
     conda
 commands =
-    sh -c 'if conda info --envs | grep -q poli__ax; then echo "poli__ax already exists"; else conda env create -f ./src/poli_baselines/core/utils/ax/environment.ax.yml; fi'
+    sh -c 'if conda info --envs | grep -q poli__ax; then echo "poli__ax already exists"; else conda env create -f ./src/poli_baselines/core/utils/ax/environment.ax.small.yml; fi'
     sh -c "conda run -n poli__ax python -m pip install -e ."
     sh -c "conda run -n poli__ax python -m pip install pytest"
     conda run -n poli__ax pytest {tty:--color=yes} -v src/poli_baselines/tests/solvers/bayesian_optimization/test_ax_solvers.py  {posargs}
@@ -72,7 +72,7 @@ allowlist_externals =
     sh
     conda
 commands =
-    sh -c 'if conda info --envs | grep -q poli__alebo; then echo "poli__alebo already exists"; else conda env create -f ./src/poli_baselines/solvers/bayesian_optimization/alebo/environment.alebo.yml; fi'
+    sh -c 'if conda info --envs | grep -q poli__alebo; then echo "poli__alebo already exists"; else conda env create -f ./src/poli_baselines/solvers/bayesian_optimization/alebo/environment.alebo.small.yml; fi'
     sh -c "conda run -n poli__alebo python -m pip install -e ."
     sh -c "conda run -n poli__alebo python -m pip install pytest"
     conda run -n poli__alebo pytest {tty:--color=yes} -v src/poli_baselines/tests/solvers/bayesian_optimization/test_alebo.py  {posargs}
@@ -85,7 +85,7 @@ allowlist_externals =
     sh
     conda
 commands =
-    sh -c 'if conda info --envs | grep -q poli__pr; then echo "poli__pr already exists"; else conda env create -f ./src/poli_baselines/solvers/bayesian_optimization/pr/environment.pr.yml; fi'
+    sh -c 'if conda info --envs | grep -q poli__pr; then echo "poli__pr already exists"; else conda env create -f ./src/poli_baselines/solvers/bayesian_optimization/pr/environment.pr.small.yml; fi'
     sh -c "conda run -n poli__pr python -m pip install -e ."
     sh -c "conda run -n poli__pr python -m pip install pytest"
     conda run -n poli__pr pytest {tty:--color=yes} -v src/poli_baselines/tests/solvers/bayesian_optimization/test_pr.py  {posargs}