build: add optimization optional deps

mdtanker · Nov 30, 2023 · 871a870 · 871a870
1 parent 203c1c8
commit 871a870
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 16 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -55,7 +55,6 @@
     # numba_progress
     # "tqdm": ("https://tqdm.github.io/", None),
     "pygmt": ("https://www.pygmt.org/latest/", None),
-    "optuna": ("https://optuna.readthedocs.io/en/stable/", None),
     #
     # Viz deps
     #
@@ -64,6 +63,10 @@
     # ipywidgets
     "matplotlib": ("https://matplotlib.org/", None),
     "seaborn": ("https://seaborn.pydata.org/", None),
+    #
+    # Opti deps
+    #
+    "optuna": ("https://optuna.readthedocs.io/en/stable/", None),
 }
 
 nitpick_ignore = [

diff --git a/env/RTD_env.yml b/env/RTD_env.yml
@@ -22,8 +22,11 @@ dependencies:
   - tqdm
   - pygmt
   - dask
+  # opti
   - optuna>=3.1.0
   - botorch>=0.4.0
+  - psutil
+  - joblib
   # docs
   - sphinx>=4.0
   - myst-parser
@@ -33,8 +36,5 @@ dependencies:
   - sphinx-design
   - nbsphinx
   - pip:
-      - numba-progress
       - tqdm_joblib
-      - psutil
-      - joblib
       - ../.
diff --git a/env/test_env.yml b/env/test_env.yml
@@ -22,8 +22,11 @@ dependencies:
   - tqdm
   - pygmt
   - dask
+  # opti
   - optuna>=3.1.0
   - botorch>=0.4.0
+  - psutil
+  - joblib
   # viz
   - pyvista
   - trame
@@ -42,8 +45,5 @@ dependencies:
   - sphinx-design
   - nbsphinx
   - pip:
-      - numba-progress
       - tqdm_joblib
-      - psutil
-      - joblib
       - ../.
diff --git a/noxfile.py b/noxfile.py
@@ -80,7 +80,7 @@ def docs(session: nox.Session) -> None:
 
     extra_installs = ["sphinx-autobuild"] if args.serve else []
 
-    session.install("-e.[docs,viz]", *extra_installs)
+    session.install("-e.[opti,docs,viz]", *extra_installs)
     session.chdir("docs")
 
     if args.builder == "linkcheck":

diff --git a/pyproject.toml b/pyproject.toml
@@ -45,17 +45,16 @@ dependencies = [
   "tqdm",
   "pygmt",
   "dask",
-  "tqdm_joblib",
-  "psutil",
-  "joblib",
-  "optuna>=3.1.0", # need JournalStorage
-  "botorch>=0.4.0",
-
-#  "geopandas",
-#  "boule",
 ]
 
 [project.optional-dependencies]
+opti = [
+  "optuna>=3.1.0", # need JournalStorage
+  "botorch>=0.4.0",
+  "joblib",
+  "psutil",
+  "tqdm_joblib",
+]
 viz = [
   "pyvista",
   "trame",