(fix): use ott-jax[neural]

theislab · May 22, 2024 · ea3dc93 · ea3dc93
1 parent 98177c1
commit ea3dc93
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -55,11 +55,9 @@ dependencies = [
     "scanpy>=1.9.3",
     "wrapt>=1.13.2",
     "docrep>=0.3.2",
-    "ott-jax>=0.4.6",
+    "ott-jax[neural]>=0.4.6",
     "cloudpickle>=2.2.0",
     "rich>=13.5",
-    "optax>=0.1.7",
-    "flax>=0.7.1",
 ]
 
 [project.optional-dependencies]