From 8ffac1da5ecab1d96d7bf718e324752821374d0e Mon Sep 17 00:00:00 2001 From: quattro Date: Tue, 13 Feb 2024 10:37:33 -0800 Subject: [PATCH] added other missing requirements --- requirements.txt | 2 ++ setup.cfg | 2 ++ 2 files changed, 4 insertions(+) diff --git a/requirements.txt b/requirements.txt index 6c6488b..f51d300 100644 --- 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 index 95ee355..0e8a79d 100644 --- a/setup.cfg +++ b/setup.cfg @@ -39,6 +39,8 @@ install_requires = pandas plum-dispatch scikit-learn + equinox + optax [options.packages.find] exclude =