From f412a2637b6ae83f9fc1ecaa60b5205d4d43507d Mon Sep 17 00:00:00 2001 From: Raphael Jin Date: Fri, 21 Jun 2024 21:36:12 +0000 Subject: [PATCH] [Deploy] Nit. --- .../scheduler/model_scheduler/device_server_constants.py | 2 +- .../scheduler/model_scheduler/master_protocol_manager.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/fedml/computing/scheduler/model_scheduler/device_server_constants.py b/python/fedml/computing/scheduler/model_scheduler/device_server_constants.py index c41b150bc2..00f0fe73bf 100644 --- a/python/fedml/computing/scheduler/model_scheduler/device_server_constants.py +++ b/python/fedml/computing/scheduler/model_scheduler/device_server_constants.py @@ -109,7 +109,7 @@ class ServerConstants(object): INFERENCE_REQUEST_TIMEOUT_KEY = "request_timeout_sec" INFERENCE_REQUEST_TIMEOUT_DEFAULT = 30 - USER_ENCRYPTED_API_KEY = "user_encrypted_api_key" + USER_ENCRYPTED_API_KEY = "encrypted_api_key" # -----End----- MODEL_DEPLOYMENT_STAGE1 = {"index": 1, "text": "ReceivedRequest"} diff --git a/python/fedml/computing/scheduler/model_scheduler/master_protocol_manager.py b/python/fedml/computing/scheduler/model_scheduler/master_protocol_manager.py index 5e16d5a02a..9e0d51b588 100755 --- a/python/fedml/computing/scheduler/model_scheduler/master_protocol_manager.py +++ b/python/fedml/computing/scheduler/model_scheduler/master_protocol_manager.py @@ -164,7 +164,7 @@ def callback_start_deployment(self, topic, payload): target_queries_per_replica = request_json.get("target_queries_per_replica", 10) aggregation_window_size_seconds = request_json.get("aggregation_window_size_seconds", 60) scale_down_delay_seconds = request_json.get("scale_down_delay_seconds", 120) - user_encrypted_api_key = request_json.get("encrypted_api_key", "") + user_encrypted_api_key = request_json.get(ServerConstants.USER_ENCRYPTED_API_KEY, "") model_config = request_json["model_config"] model_name = model_config["model_name"]