in code/workflow/implementations/autopilot/bp_xai_analysis_stage.py [0:0]
def monitor_job_status(job_name, context) :
sleep_time = 60
while True:
status = sm.describe_processing_job(ProcessingJobName = job_name)["ProcessingJobStatus"]
if status == 'Completed' :
break;
elif status in ('Failed', 'Stopped') :
break;
else :
if context.get_remaining_time_in_millis() > 2000*sleep_time :
sleep(sleep_time)
else :
raise TaskTimedOut("Task timed out.")
return {"status":status}