`airflow.contrib.operators.sagemaker_base_operator`¶

Module Contents¶

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

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'][source]¶

template_ext = [][source]¶

ui_color = #ededed[source]¶

integer_fields = [][source]¶

parse_integer(self, config, field)[source]¶

parse_config_integers(self)[source]¶

expand_role(self)[source]¶

preprocess_config(self)[source]¶

execute(self, context)[source]¶