@@ -54,6 +54,12 @@ def register_request_and_wait_for_response(
5454 return self .infer .get_timeout_response (
5555 unique_id , is_compressed , input_type , is_client_timeout = True
5656 )
57+ except Exception as e :
58+ _utils .logger .exception (e , exc_info = True )
59+ _utils .logger .error (f"Error registering request and waiting for response: { e } " )
60+ return self .infer .get_timeout_response (
61+ unique_id , is_compressed , input_type , is_client_timeout = True
62+ )
5763 finally :
5864 with self .lock :
5965 self .pending_requests .pop (unique_id , None )
@@ -388,7 +394,7 @@ def on_get(self, req, resp):
388394 }
389395 return
390396
391- elif fail_if_requests_older_than_x_seconds_pending_param :
397+ if fail_if_requests_older_than_x_seconds_pending_param :
392398 if _utils .check_if_requests_older_than_x_seconds_pending (
393399 int (fail_if_requests_older_than_x_seconds_pending_param )
394400 ):
@@ -398,7 +404,7 @@ def on_get(self, req, resp):
398404 }
399405 return
400406
401- elif fail_if_up_time_more_than_x_seconds_param :
407+ if fail_if_up_time_more_than_x_seconds_param :
402408 if time .time () - Infer .started_at_time > int (
403409 fail_if_up_time_more_than_x_seconds_param
404410 ):
@@ -408,7 +414,7 @@ def on_get(self, req, resp):
408414 }
409415 return
410416
411- elif fail_if_requests_timedout_in_last_x_seconds_is_more_than_y_param :
417+ if fail_if_requests_timedout_in_last_x_seconds_is_more_than_y_param :
412418 (
413419 x ,
414420 y ,
0 commit comments