Skip to content

Commit 9d5b54f

Browse files
committed
Nit: Update naming
1 parent 3792255 commit 9d5b54f

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

python/fedml/computing/scheduler/model_scheduler/device_model_deployment.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -305,15 +305,15 @@ def start_deployment(end_point_id, end_point_name, model_id, model_version,
305305
# TODO: Make sure no competition for each replica in a single deployment
306306
if exist_container_obj is not None:
307307
client.api.remove_container(exist_container_obj.id, v=True, force=True)
308-
device_requests = {}
308+
device_mapping = {}
309309
if no_real_gpu_allocation is not None:
310310
use_gpu = not no_real_gpu_allocation
311311
if use_gpu:
312312
logging.info("Number of GPUs: {}".format(num_gpus))
313-
device_requests = HardwareUtil.get_docker_gpu_device_mapping(gpu_ids, num_gpus)
314-
logging.info(f"device_requests: {device_requests}")
313+
device_mapping = HardwareUtil.get_docker_gpu_device_mapping(gpu_ids, num_gpus)
314+
logging.info(f"device_mapping: {device_mapping}")
315315

316-
host_config_dict.update(device_requests)
316+
host_config_dict.update(device_mapping)
317317

318318
# Environment variables
319319
if not enable_custom_image:

0 commit comments

Comments
 (0)