Switched pytest to default dependencies

srigas · Jan 6, 2025 · 0041fbf · 0041fbf
1 parent 7966514
commit 0041fbf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -33,7 +33,8 @@ dependencies = [
     "scipy==1.14.1",
     "jax[cpu]==0.4.35",
     "optax==0.2.4",
-    "flax==0.10.2"
+    "flax==0.10.2",
+    "pytest==8.3.4",
 ]
 
 
@@ -47,7 +48,6 @@ gpu = [
     "jax[cuda12]==0.4.35"
 ]
 doc = [
-    "pytest==8.3.4",
     "jupyterlab==4.3.0",
     "matplotlib==3.9.2",
     "scikit-learn==1.5.2",