:mod:`airflow.contrib.sensors.sagemaker_training_sensor` ======================================================== .. py:module:: airflow.contrib.sensors.sagemaker_training_sensor Module Contents --------------- .. py:class:: SageMakerTrainingSensor(job_name, print_log=True, *args, **kwargs) Bases::class:`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. :param job_name: name of the SageMaker training job to check the state of :type job_name: str :param print_log: if the operator should print the cloudwatch log :type print_log: bool .. attribute:: template_fields :annotation: = ['job_name'] .. attribute:: template_ext :annotation: = [] .. method:: init_log_resource(self, hook) .. method:: non_terminal_states(self) .. method:: failed_states(self) .. method:: get_sagemaker_response(self) .. method:: get_failed_reason_from_response(self, response) .. method:: state_from_response(self, response)