SageMakerTrainingSensor
AmazonPolls the training job until it reaches a terminal state. Raises an AirflowException with the failure reason if a failed state is reached.
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_nameRequiredName of the training job to watch.
print_logPrints the cloudwatch log if True; Defaults to True.
Documentation
Polls the training job until it reaches a terminal state. Raises an AirflowException with the failure reason if a failed state is reached.
See also
For more information on how to use this sensor, take a look at the guide: Wait on an Amazon SageMaker training job state