airflow.providers.amazon.aws.operators.sagemaker_base

Module Contents

class airflow.providers.amazon.aws.operators.sagemaker_base.SageMakerBaseOperator(*, config: dict, aws_conn_id: str = 'aws_default', **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 :Iterable[Iterable[str]] = [][source]
parse_integer(self, config, field)[source]

Recursive method for parsing string fields holding integer values to integers.

parse_config_integers(self)[source]

Parse the integer fields of training config to integers in case the config is rendered by Jinja and all fields are str.

expand_role(self)[source]

Placeholder for calling boto3's expand_role(), which expands an IAM role name into an ARN.

preprocess_config(self)[source]

Process the config into a usable form.

execute(self, context)[source]
hook(self)[source]

Return SageMakerHook

Was this entry helpful?