airflow.contrib.operators.sagemaker_endpoint_config_operator

Module Contents

class airflow.contrib.operators.sagemaker_endpoint_config_operator.SageMakerEndpointConfigOperator(config, *args, **kwargs)

Bases: airflow.contrib.operators.sagemaker_base_operator.SageMakerBaseOperator

Create a SageMaker endpoint config.

This operator returns The ARN of the endpoint config created in Amazon SageMaker

Parameters

• config (dict) –

  The configuration necessary to create an endpoint config.

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

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

integer_fields = [['ProductionVariants', 'InitialInstanceCount']]

execute(self, context)