airflow.contrib.operators.sagemaker_model_operator

Module Contents

class airflow.contrib.operators.sagemaker_model_operator.SageMakerModelOperator(config, *args, **kwargs)

Bases:airflow.contrib.operators.sagemaker_base_operator.SageMakerBaseOperator

Create a SageMaker model.

This operator returns The ARN of the model created in Amazon SageMaker

Parameters

• config (dict) –

  The configuration necessary to create a model.

  For details of the configuration parameter see SageMaker.Client.create_model()

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

expand_role(self)

execute(self, context)