[CoreEngine] add the requirements.txt for hello world example.

FedML-AI · Aug 14, 2023 · ccab13a · ccab13a
1 parent 9e1991d
commit ccab13a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/python/fedml/__init__.py b/python/fedml/__init__.py
@@ -25,7 +25,7 @@
 _global_training_type = None
 _global_comm_backend = None
 
-__version__ = "0.8.8a11"
+__version__ = "0.8.8a12"
 
 
 def init(args=None):

diff --git a/python/fedml/cli/scheduler/hello_world/requirements.txt b/python/fedml/cli/scheduler/hello_world/requirements.txt
@@ -0,0 +1,3 @@
+numpy==1.24.2
+torch==2.0.0
+torchvision==0.15.1
diff --git a/python/setup.py b/python/setup.py
@@ -91,7 +91,7 @@ def finalize_options(self):
 
 setup(
     name="fedml",
-    version="0.8.8a11",
+    version="0.8.8a12",
     author="FedML Team",
     author_email="[email protected]",
     description="A research and production integrated edge-cloud library for "