Hotfix mqtt timeout inference constant after refactoring.

fedml-dimitris · fedml-dimitris · commit 1cc15525e2b8 · 2024-06-04T04:38:30.000-04:00
diff --git a/python/fedml/computing/scheduler/model_scheduler/device_mqtt_inference_protocol.py b/python/fedml/computing/scheduler/model_scheduler/device_mqtt_inference_protocol.py
@@ -105,7 +105,7 @@ def run_mqtt_inference_with_request(
             only_do_health_check=only_do_health_check, timeout=timeout
         )
 
-        allowed_inference_timeout = SchedulerConstants.MQTT_INFERENCE_TIMEOUT if timeout is None else timeout
+        allowed_inference_timeout = timeout if timeout else -1
         sleep_time_interval = 0.05
         total_sleep_time = 0
         while True:

Original file line number	Diff line number	Diff line change
`@@ -105,7 +105,7 @@ def run_mqtt_inference_with_request(`
`105`	`105`	`only_do_health_check=only_do_health_check, timeout=timeout`
`106`	`106`	`)`
`107`	`107`
`108`		`- allowed_inference_timeout = SchedulerConstants.MQTT_INFERENCE_TIMEOUT if timeout is None else timeout`
	`108`	`+ allowed_inference_timeout = timeout if timeout else -1`
`109`	`109`	`sleep_time_interval = 0.05`
`110`	`110`	`total_sleep_time = 0`
`111`	`111`	`while True:`