airflow.contrib.sensors.sagemaker_training_sensor
¶
Module Contents¶
-
class
airflow.contrib.sensors.sagemaker_training_sensor.
SageMakerTrainingSensor
(job_name, print_log=True, *args, **kwargs)[source]¶ Bases:
airflow.contrib.sensors.sagemaker_base_sensor.SageMakerBaseSensor
Asks for the state of the training state until it reaches a terminal state. If it fails the sensor errors, failing the task.
- Parameters