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