def is_processing_job_ready()

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