airflow.contrib.operators.sagemaker_base_operator

Module Contents

class airflow.contrib.operators.sagemaker_base_operator.SageMakerBaseOperator(config, aws_conn_id='aws_default', *args, **kwargs)

Bases: airflow.models.BaseOperator

This is the base operator for all SageMaker operators.

Parameters

• config (dict) – The configuration necessary to start a training job (templated)

• aws_conn_id (str) – The AWS connection ID to use.

template_fields = ['config']

template_ext = []

ui_color = #ededed

integer_fields :Iterable[Iterable[str]] = []

parse_integer(self, config, field)

parse_config_integers(self)

expand_role(self)

preprocess_config(self)

execute(self, context)