SageMakerEndpointOperator

Amazon

Create a SageMaker endpoint.

View Source

Last Updated: May. 7, 2021

Access Instructions

Install the Amazon provider package into your Airflow environment.

Import the module into your DAG file and instantiate it with your desired params.

Parameters

configdictThe configuration necessary to create an endpoint.If you need to create a SageMaker endpoint based on an existed SageMaker model and an existed SageMaker endpoint config:config = endpoint_configuration;If you need to create all of SageMaker model, SageMaker endpoint-config and SageMaker endpoint:config = { 'Model': model_configuration, 'EndpointConfig': endpoint_config_configuration, 'Endpoint': endpoint_configuration }For details of the configuration parameter of model_configuration see :py:meth:`SageMaker.Client.create_model`No role entry for "py:meth" in module "docutils.parsers.rst.languages.en". Trying "py:meth" as canonical role name.Unknown interpreted text role "py:meth".For details of the configuration parameter of endpoint_config_configuration see :py:meth:`SageMaker.Client.create_endpoint_config`No role entry for "py:meth" in module "docutils.parsers.rst.languages.en". Trying "py:meth" as canonical role name.Unknown interpreted text role "py:meth".For details of the configuration parameter of endpoint_configuration see :py:meth:`SageMaker.Client.create_endpoint`No role entry for "py:meth" in module "docutils.parsers.rst.languages.en". Trying "py:meth" as canonical role name.Unknown interpreted text role "py:meth".
aws_conn_idstrThe AWS connection ID to use.
wait_for_completionboolWhether the operator should wait until the endpoint creation finishes.
check_intervalintIf wait is set to True, this is the time interval, in seconds, that this operation waits before polling the status of the endpoint creation.
max_ingestion_timeintIf wait is set to True, this operation fails if the endpoint creation doesn't finish within max_ingestion_time seconds. If you set this parameter to None it never times out.
operationstrWhether to create an endpoint or update an endpoint. Must be either 'create or 'update'.

Documentation

Create a SageMaker endpoint.

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

Example DAGs

Improve this module by creating an example DAG.

View Source
  1. Add an `example_dags` directory to the top-level source of the provider package with an empty `__init__.py` file.
  2. Add your DAG to this directory. Be sure to include a well-written and descriptive docstring
  3. Create a pull request against the source code. Once the package gets released, your DAG will show up on the Registry.

Was this page helpful?