Fix dependencies

scbirlab · Oct 14, 2024 · 99c3396 · 99c3396
1 parent 2199d18
commit 99c3396
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -40,16 +40,24 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
+fit_base = [
+  "scipy",
+  "matplotlib",
+]
 fit = [
+  "crispio[fit_base]",
+  "jax",
+  "optax",
+]
+fit_cuda12 = [
+  "crispio[fit_base]",
   "jax[cuda12]",
   "optax",
-  "scipy",
-  "matplotlib",
 ]
-fit_cuda_local = [
+fit_cuda12_local = [
+  "crispio[fit_base]",
   "jax[cuda12_local]",
   "optax",
-  "scipy",
 ]
 
 [project.urls]