Fix circle CI

secretflow · Oct 23, 2024 · c376150 · c376150
1 parent 4c01c51
commit c376150
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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