@@ -865,17 +865,17 @@ def _wait_for_invocation_jobs(ctx, gi, invocation_id, polling_backoff=0, early_t
865865 def state_func ():
866866 return _retry_on_timeouts (ctx , gi , lambda gi : gi .jobs .get_jobs (invocation_id = invocation_id ))
867867
868- return _wait_on_state (ctx , state_func , polling_backoff , early_termination = early_termination )
868+ return _wait_on_state (state_func , polling_backoff , early_termination = early_termination )
869869
870870
871871def _wait_for_job (gi , job_id , timeout = None ):
872872 def state_func ():
873873 return gi .jobs .show_job (job_id , full_details = True )
874874
875- return _wait_on_state (ctx , state_func , timeout = timeout )
875+ return _wait_on_state (state_func , timeout = timeout )
876876
877877
878- def _wait_on_state (ctx , state_func , polling_backoff = 0 , timeout = None , early_termination = True ):
878+ def _wait_on_state (state_func , polling_backoff = 0 , timeout = None , early_termination = True ):
879879 def get_state ():
880880 response = state_func ()
881881 if not isinstance (response , list ):
@@ -902,7 +902,6 @@ def get_state():
902902 for terminal_state in hierarchical_fail_states :
903903 if terminal_state in current_states :
904904 # If we got here something has failed and we can return (early)
905- ctx .log (f"Early termination." )
906905 return terminal_state
907906 if current_non_terminal_states :
908907 return None
0 commit comments