From 96aa730d29ddd8399a886a54c15e3ce4b3f4ba6d Mon Sep 17 00:00:00 2001
From: SBouchard01 <simonbouchard47@gmail.com>
Date: Mon, 8 Jun 2026 10:33:54 -0700
Subject: [PATCH] feat: move git-installed deps to cosmodesi optional
 dependencies

---
 pyproject.toml | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index d95d9c7..a00d6cb 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -26,12 +26,10 @@ dependencies = [
     "torch",
     "lightning",
     "pytorch-lightning",
-    "jax",
     "flax",
     "joblib",
     "optuna",
     "deprecation",
-    'cosmoprimo[class,camb,astropy,extras] @ git+https://github.com/cosmodesi/cosmoprimo',
     "optax",
     "wandb",
     "mpi4py",
@@ -39,13 +37,17 @@ dependencies = [
 
 
 [project.optional-dependencies]
+cosmodesi = [ # Dependencies existing in the cosmodesi ecosystem
+    "jax",
+    "cosmoprimo[class,camb,astropy,extras] @ git+https://github.com/cosmodesi/cosmoprimo",
+    "desilike @ git+https://github.com/cosmodesi/desilike",
+]
 inference = [
     "matplotlib",
     "getdist",
     "tabulate",
-    "Py-BOBYQA",
-    "desilike @ git+https://github.com/cosmodesi/desilike",
     "h5py",
+    "Py-BOBYQA",
     "dynesty",
     "emcee",
     "numpyro",