Skip to content

Commit 2fcf57d

Browse files
committed
Add check as device_mapping can be None
1 parent 9d5b54f commit 2fcf57d

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,8 @@ def start_deployment(end_point_id, end_point_name, model_id, model_version,
313313
device_mapping = HardwareUtil.get_docker_gpu_device_mapping(gpu_ids, num_gpus)
314314
logging.info(f"device_mapping: {device_mapping}")
315315

316-
host_config_dict.update(device_mapping)
316+
if device_mapping:
317+
host_config_dict.update(device_mapping)
317318

318319
# Environment variables
319320
if not enable_custom_image:

0 commit comments

Comments
 (0)