fix(req): update eggroll requirements

Signed-off-by: weiwee <wbwmat@gmail.com>
FederatedAI · Jan 31, 2023 · fd05551 · fd05551
1 parent b2e4345
commit fd05551
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/python/requirements-eggroll.txt b/python/requirements-eggroll.txt
@@ -1,6 +1,12 @@
 grpcio==1.46.3
 grpcio-tools==1.46.3
-numba==0.53.0
+numba==0.56.4
 protobuf==3.19.6
 pyarrow==6.0.1
 mmh3==3.0.0
+cachetools>=3.0.0
+cloudpickle==2.1.0
+llvmlite==0.36.0
+psutil>=5.7.0
+python-rocksdb==0.7.0
+pystack-debugger>=0.9.0
diff --git a/python/setup.py b/python/setup.py
@@ -34,6 +34,12 @@
         "protobuf==3.19.6",
         "pyarrow==6.0.1",
         "mmh3==3.0.0",
+        "cachetools>=3.0.0",
+        "cloudpickle==2.1.0",
+        "llvmlite==0.36.0",
+        "psutil>=5.7.0",
+        "python-rocksdb==0.7.0",
+        "pystack-debugger>=0.9.0",
     ],
     "all": ["pyfate[rabbitmq,pulsar,spark,eggroll]"],
 }