in custom_resource/sagemaker_suggest_baseline.py [0:0]
def is_processing_job_ready(processing_job_name):
is_ready = False
processing_job = sm.describe_processing_job(ProcessingJobName=processing_job_name)
status = processing_job["ProcessingJobStatus"]
if status == "Stopped" or status == "Completed":
logger.info("Processing Job (%s) is %s", processing_job_name, status)
is_ready = True
elif status == "InProgress" or status == "Stopping":
logger.info(
"Processing Job (%s) still in progress, waiting and polling again...",
processing_job_name,
)
else:
raise Exception(
"Processing Job ({}) has unexpected status: {}".format(processing_job_name, status)
)
return is_ready