airflow.contrib.operators.sagemaker_model_operator

Module Contents

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

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)[source]
execute(self, context)[source]