added other missing requirements

mancusolab · Feb 13, 2024 · 8ffac1d · 8ffac1d
1 parent 1b29366
commit 8ffac1d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,8 @@
+equinox
 jax
 jaxlib
 numpy
+optax
 pandas
 plum-dispatch
 scikit-learn
diff --git a/setup.cfg b/setup.cfg
@@ -39,6 +39,8 @@ install_requires =
     pandas
     plum-dispatch
     scikit-learn
+    equinox
+    optax
 
 [options.packages.find]
 exclude =