SageMakerTrainingSensor

Amazon

Asks for the state of the training state until it reaches a terminal state. If it fails the sensor errors, failing the task.

View on GitHub

Last Updated: May. 7, 2021

Access Instructions

Install the Amazon provider package into your Airflow environment.

Import the module into your DAG file and instantiate it with your desired params.

Parameters

job_nameRequiredstrname of the SageMaker training job to check the state of
print_logboolif the operator should print the cloudwatch log

Documentation

Asks for the state of the training state until it reaches a terminal state. If it fails the sensor errors, failing the task.

Was this page helpful?