Skip to content

Commit 8530973

Browse files
Merge pull request #2165 from FedML-AI/raphael/refactor-container-deployment
[Deploy] Remove unnecessary logic
2 parents 34fdba0 + 9194f84 commit 8530973

6 files changed

Lines changed: 57 additions & 357 deletions

File tree

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

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -369,7 +369,7 @@ def get_idle_device(self,
369369
if "model_status" in result_payload and result_payload["model_status"] == "DEPLOYED":
370370
idle_device_list.append({"device_id": device_id, "end_point_id": end_point_id})
371371

372-
logging.info(f"{len(idle_device_list)} devices this model has on it: {idle_device_list}")
372+
logging.debug(f"{len(idle_device_list)} devices this model has on it: {idle_device_list}")
373373

374374
if len(idle_device_list) <= 0:
375375
return None, None
@@ -398,7 +398,7 @@ def get_idle_device(self,
398398
logging.info("Inference Device selection Failed:")
399399
logging.info(e)
400400

401-
logging.info(f"Using Round Robin, the device index is {selected_device_index}")
401+
logging.debug(f"Using Round Robin, the device index is {selected_device_index}")
402402
idle_device_dict = idle_device_list[selected_device_index]
403403

404404
# Note that within the same endpoint_id, there could be one device with multiple same models
@@ -411,7 +411,7 @@ def get_idle_device(self,
411411
# Find deployment result from the target idle device.
412412
try:
413413
for result_item in result_list:
414-
logging.info("enter the for loop")
414+
logging.debug("enter the for loop")
415415
device_id, _, result_payload = self.get_result_item_info(result_item)
416416
found_end_point_id = result_payload["end_point_id"]
417417
found_end_point_name = result_payload["end_point_name"]
@@ -425,7 +425,7 @@ def get_idle_device(self,
425425
if same_model_device_rank > 0:
426426
same_model_device_rank -= 1
427427
continue
428-
logging.info(f"The chosen device is {device_id}")
428+
logging.debug(f"The chosen device is {device_id}")
429429
return result_payload, device_id
430430
except Exception as e:
431431
logging.info(str(e))

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

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414

1515
from fedml.core.common.singleton import Singleton
1616
from fedml.computing.scheduler.model_scheduler.modelops_configs import ModelOpsConfigs
17-
from fedml.computing.scheduler.model_scheduler.device_model_deployment import get_model_info
1817
from fedml.computing.scheduler.model_scheduler.device_server_constants import ServerConstants
1918
from fedml.computing.scheduler.model_scheduler.device_model_object import FedMLModelList, FedMLEndpointDetail
2019
from fedml.computing.scheduler.model_scheduler.device_client_constants import ClientConstants

0 commit comments

Comments
 (0)