:mod:`airflow.contrib.operators.sagemaker_base_operator`
========================================================

.. py:module:: airflow.contrib.operators.sagemaker_base_operator


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

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

   Bases: :class:`airflow.models.BaseOperator`

   This is the base operator for all SageMaker operators.

   :param config: The configuration necessary to start a training job (templated)
   :type config: dict
   :param aws_conn_id: The AWS connection ID to use.
   :type aws_conn_id: str

   .. attribute:: template_fields
      :annotation: = ['config']

      
   .. attribute:: template_ext
      :annotation: = []

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

      
   .. attribute:: integer_fields
      :annotation: :Iterable[Iterable[str]] = []

      
   .. method:: parse_integer(self, config, field)


   .. method:: parse_config_integers(self)


   .. method:: expand_role(self)


   .. method:: preprocess_config(self)


   .. method:: execute(self, context)