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]