make neural dependencies optional

theislab · Oct 25, 2023 · 517db5f · 517db5f
1 parent 4e964e5
commit 517db5f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -80,6 +80,8 @@ test = [
     "pytest-mock>=3.5.0",
     "pytest-cov>=4",
     "coverage[toml]>=7",
+    "optax>=0.1.7",
+    "flax>=0.7.1", 
 ]
 docs = [
     "sphinx>=5.1.1",
@@ -270,7 +272,7 @@ env_list = lint-code,py{3.8,3.9,3.10,3.11}
 skip_missing_interpreters = true
 
 [testenv]
-extras = [test, neural]
+extras = [test]
 pass_env = PYTEST_*,CI
 commands =
     python -m pytest {tty:--color=yes} {posargs: \