Fixed a race condition and client privacy.json issue. (#1164) (#1172)

NVIDIA · Dec 1, 2022 · 01f3bb4 · 01f3bb4
1 parent 311ab02
commit 01f3bb4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/nvflare/private/fed/app/fl_conf.py b/nvflare/private/fed/app/fl_conf.py
@@ -33,7 +33,7 @@
 from .fl_app_validator import FLAppValidator
 
 FL_PACKAGES = ["nvflare"]
-FL_MODULES = ["server", "client", "app"]
+FL_MODULES = ["server", "client", "app_common", "private"]
 
 
 class FLServerStarterConfiger(JsonConfigurator):

diff --git a/nvflare/private/fed/server/fed_server.py b/nvflare/private/fed/server/fed_server.py
@@ -439,7 +439,7 @@ def GetTask(self, request, context):
                 return task
 
     def _process_task_request(self, client, fl_ctx, shared_fl_ctx):
-        task_name = SpecialTaskName.END_RUN
+        task_name = SpecialTaskName.TRY_AGAIN
         task_id = ""
         shareable = None
         try: