| 55 | |
| 56 | |
| 57 | class FedMLClientRunner: |
| 58 | |
| 59 | def __init__(self, args, edge_id=0, request_json=None, agent_config=None, run_id=0, |
| 60 | cuda_visible_gpu_ids_str=None): |
| 61 | self.model_device_server = None |
| 62 | self.model_device_client = None |
| 63 | self.run_process_event = None |
| 64 | self.run_process_event_map = dict() |
| 65 | self.run_process_completed_event = None |
| 66 | self.run_process_completed_event_map = dict() |
| 67 | self.run_process = None |
| 68 | self.run_process_map = dict() |
| 69 | self.local_api_process = None |
| 70 | self.start_request_json = None |
| 71 | self.device_status = None |
| 72 | self.current_training_status = None |
| 73 | self.mqtt_mgr = None |
| 74 | self.client_mqtt_mgr = None |
| 75 | self.client_mqtt_is_connected = False |
| 76 | self.client_mqtt_lock = None |
| 77 | self.edge_id = edge_id |
| 78 | self.edge_user_name = None |
| 79 | self.edge_extra_url = None |
| 80 | self.run_id = run_id |
| 81 | self.unique_device_id = None |
| 82 | self.args = args |
| 83 | self.request_json = request_json |
| 84 | self.version = args.version |
| 85 | self.device_id = args.device_id |
| 86 | self.cur_dir = os.path.split(os.path.realpath(__file__))[0] |
| 87 | if args.current_running_dir is not None: |
| 88 | self.cur_dir = args.current_running_dir |
| 89 | self.sudo_cmd = "" |
| 90 | self.is_mac = False |
| 91 | if platform.system() == "Darwin": |
| 92 | self.is_mac = True |
| 93 | |
| 94 | self.agent_config = agent_config |
| 95 | self.fedml_data_base_package_dir = os.path.join("/", "fedml", "data") |
| 96 | self.fedml_data_local_package_dir = os.path.join("/", "fedml", "fedml-package", "fedml", "data") |
| 97 | self.fedml_data_dir = self.fedml_data_base_package_dir |
| 98 | self.fedml_config_dir = os.path.join("/", "fedml", "conf") |
| 99 | |
| 100 | self.FEDML_DYNAMIC_CONSTRAIN_VARIABLES = { |
| 101 | "${FEDSYS.RUN_ID}": "", |
| 102 | "${FEDSYS.PRIVATE_LOCAL_DATA}": "", |
| 103 | "${FEDSYS.CLIENT_ID_LIST}": "", |
| 104 | "${FEDSYS.SYNTHETIC_DATA_URL}": "", |
| 105 | "${FEDSYS.IS_USING_LOCAL_DATA}": "", |
| 106 | "${FEDSYS.CLIENT_NUM}": "", |
| 107 | "${FEDSYS.CLIENT_INDEX}": "", |
| 108 | "${FEDSYS.CLIENT_OBJECT_LIST}": "", |
| 109 | "${FEDSYS.LOG_SERVER_URL}": "", |
| 110 | } |
| 111 | |
| 112 | self.mlops_metrics = None |
| 113 | self.client_active_list = dict() |
| 114 | self.ntp_offset = MLOpsUtils.get_ntp_offset() |
no outgoing calls
no test coverage detected