revert requirements.txt

secretflow · Jan 3, 2025 · f617bdf · f617bdf
1 parent d8058e8
commit f617bdf
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 0 deletions.
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -0,0 +1,30 @@
+# basic deps
+grpcio==1.66.0
+numpy>=1.22.0, <2 # FIXME: for SF compatibility
+protobuf==5.27.3
+cloudpickle>=2.0.0
+multiprocess>=0.70.12.2
+cachetools>=5.0.0
+jax[cpu]>=0.4.16, <=0.4.34 # FIXME: Jax 0.4.26+ select perf issue
+termcolor>=2.0.0
+pandas>=1.4.2
+flax<0.10.0
+scikit-learn<1.6.0
+# for tests
+absl-py>=1.1.0
+tensorflow-cpu>=2.12.0; sys_platform == "linux" and platform_machine == 'x86_64'
+h5py!=3.11.0; platform_machine == 'aarch64'
+
+# for examples
+dm-haiku
+plotnine
+jraph
+optax
+torch==2.3.0
+torch_xla==2.3.0
+torchvision
+tensorflow_datasets
+keras
+setuptools<71.0.0 # https://github.com/pypa/setuptools/issues/4487
+transformers
+datasets
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,8 @@
+grpcio==1.66.0
+numpy>=1.22.0, <2 # FIXME: for SF compatibility
+protobuf==5.27.3
+cloudpickle>=2.0.0
+multiprocess>=0.70.12.2
+cachetools>=5.0.0
+jax[cpu]>=0.4.16, <=0.4.34 # FIXME: Jax 0.4.26+ select perf issue
+termcolor>=2.0.0