Skip to content

Commit

Permalink
Fix a bug in dashboard that server local resource file was not genera…
Browse files Browse the repository at this point in the history
…ted (NVIDIA#2959)

correctly
  • Loading branch information
IsaacYangSLA authored Sep 24, 2024
1 parent afad3d2 commit 3dc3d3c
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions nvflare/lighter/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -280,8 +280,10 @@ def _write_local(type, dest_dir, template, capacity=""):
template["default_authz"],
"t",
)
resources = json.loads(template["local_client_resources"])
if type == "client":
if type == "server":
resources = json.loads(template["local_server_resources"])
elif type == "client":
resources = json.loads(template["local_client_resources"])
for component in resources["components"]:
if "nvflare.app_common.resource_managers.gpu_resource_manager.GPUResourceManager" == component["path"]:
component["args"] = json.loads(capacity)
Expand Down

0 comments on commit 3dc3d3c

Please sign in to comment.