:mod:`airflow.contrib.sensors.sagemaker_base_sensor`
====================================================

.. py:module:: airflow.contrib.sensors.sagemaker_base_sensor


Module Contents
---------------

.. py:class:: SageMakerBaseSensor(aws_conn_id='aws_default', *args, **kwargs)

   Bases: :class:`airflow.sensors.base_sensor_operator.BaseSensorOperator`

   Contains general sensor behavior for SageMaker.
   Subclasses should implement get_sagemaker_response()
   and state_from_response() methods.
   Subclasses should also implement NON_TERMINAL_STATES and FAILED_STATE methods.

   .. attribute:: ui_color
      :annotation: = #ededed

      
   .. method:: poke(self, context)


   .. 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)