Add back request header (NVIDIA#2440)

nvidianz · Mar 22, 2024 · 385dc3c · 385dc3c
1 parent b8d2c1a
commit 385dc3c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nvflare/private/fed/server/server_engine.py b/nvflare/private/fed/server/server_engine.py
@@ -760,6 +760,8 @@ def check_client_resources(self, job: Job, resource_reqs, fl_ctx: FLContext) ->
     def _make_message_for_check_resource(self, job, resource_requirements, fl_ctx):
         request = Message(topic=TrainingTopic.CHECK_RESOURCE, body=resource_requirements)
         request.set_header(RequestHeader.JOB_ID, job.job_id)
+        request.set_header(RequestHeader.REQUIRE_AUTHZ, "true")
+        request.set_header(RequestHeader.ADMIN_COMMAND, AdminCommandNames.CHECK_RESOURCES)
 
         set_message_security_data(request, job, fl_ctx)
         return request