(fix): need to require optax/flax

theislab · May 22, 2024 · 98177c1 · 98177c1
1 parent 0145b5b
commit 98177c1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/docs/notebooks b/docs/notebooks
diff --git a/pyproject.toml b/pyproject.toml
@@ -58,13 +58,11 @@ dependencies = [
     "ott-jax>=0.4.6",
     "cloudpickle>=2.2.0",
     "rich>=13.5",
-]
-
-[project.optional-dependencies]
-neural = [
     "optax>=0.1.7",
     "flax>=0.7.1",
 ]
+
+[project.optional-dependencies]
 spatial = [
     "squidpy>=1.2.3"
 ]
+90 −1		CONTRIBUTING.rst
+40 −35		examples/problems/100_tagged_arrays.ipynb
+68 −22		examples/problems/200_custom_cost_matrices.ipynb
+34 −15		examples/solvers/300_quad_problems_basic.ipynb
+27 −17		examples/solvers/400_quad_problems_advanced.ipynb