From bf6957024ba0783a2935e7b3d5c917f9f12fd34e Mon Sep 17 00:00:00 2001 From: Jimmy MA Date: Wed, 23 Oct 2024 20:03:21 +0800 Subject: [PATCH 1/3] Update requirements-dev.txt --- requirements-dev.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements-dev.txt b/requirements-dev.txt index 6ee814f8..d0875f91 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -1,5 +1,5 @@ pandas>=1.4.2 -flax +flax<0.10.0 scikit-learn # for tests absl-py>=1.1.0 From 4c01c519f2b9bfa4f6c4c4a3430ddb86eb554edc Mon Sep 17 00:00:00 2001 From: Jimmy MA Date: Wed, 23 Oct 2024 20:28:52 +0800 Subject: [PATCH 2/3] Update requirements-dev.txt --- requirements-dev.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements-dev.txt b/requirements-dev.txt index d0875f91..adbecd52 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -1,5 +1,5 @@ pandas>=1.4.2 -flax<0.10.0 +flax<0.8.5 scikit-learn # for tests absl-py>=1.1.0 From c3761504424f00ff18c28f65fa4b53e20311f83c Mon Sep 17 00:00:00 2001 From: Jimmy Date: Wed, 23 Oct 2024 20:49:24 +0800 Subject: [PATCH 3/3] Fix circle CI --- requirements-dev.txt | 2 +- requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements-dev.txt b/requirements-dev.txt index adbecd52..6ee814f8 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -1,5 +1,5 @@ pandas>=1.4.2 -flax<0.8.5 +flax scikit-learn # for tests absl-py>=1.1.0 diff --git a/requirements.txt b/requirements.txt index ac8fd6cc..3433c679 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,5 +4,5 @@ protobuf>=4, <5 cloudpickle>=2.0.0 multiprocess>=0.70.12.2 cachetools>=5.0.0 -jax[cpu]>=0.4.16, <=0.4.26 # FIXME: Jax 0.4.26+ select perf issue +jax[cpu]>=0.4.16, <=0.4.34 # FIXME: Jax 0.4.26+ select perf issue termcolor>=2.0.0