[CoreEngine] support to login with account id.

FedML-AI · Dec 7, 2023 · f2705e9 · f2705e9
1 parent 6a111bb
commit f2705e9
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/python/fedml/__init__.py b/python/fedml/__init__.py
@@ -34,7 +34,7 @@
 _global_training_type = None
 _global_comm_backend = None
 
-__version__ = "0.8.12a61"
+__version__ = "0.8.12a62"
 
 
 # This is the deployment environment used for different roles (RD/PM/BD/Public Developers). Potential VALUE: local, dev, test, release

diff --git a/python/fedml/cli/modules/login.py b/python/fedml/cli/modules/login.py
@@ -35,5 +35,10 @@ def fedml_login(api_key, version, compute_node, server, provider):
     fedml.set_env_version(version)
 
     api_key = api_key[0] if len(api_key) > 0 else None
-    authenticate(api_key)
+    try:
+        authenticate(api_key)
+    except SystemExit as e:
+        print(f"{str(e)}\n")
+        print(f"Maybe you are using account id to login, we will try to login with account {api_key}.")
+        pass
     fedml.api.login(api_key, compute_node, server, provider)
diff --git a/python/setup.py b/python/setup.py
@@ -94,7 +94,7 @@ def finalize_options(self):
 
 setup(
     name="fedml",
-    version="0.8.12a61",
+    version="0.8.12a62",
     author="FedML Team",
     author_email="[email protected]",
     description="A research and production integrated edge-cloud library for "