airflow.providers.amazon.aws.operators.sagemaker_model

Module Contents

class airflow.providers.amazon.aws.operators.sagemaker_model.SageMakerModelOperator(*, config, **kwargs)[source]

Bases: airflow.providers.amazon.aws.operators.sagemaker_base.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]

Was this entry helpful?