blob: c4e176f1b14bc6fa486859f4b2f409a1e7c275d5 [file] [log] [blame]
:mod:`airflow.contrib.operators.sagemaker_model_operator`
=========================================================
.. py:module:: airflow.contrib.operators.sagemaker_model_operator
Module Contents
---------------
.. py:class:: SageMakerModelOperator(config, *args, **kwargs)
Bases: :class:`airflow.contrib.operators.sagemaker_base_operator.SageMakerBaseOperator`
Create a SageMaker model.
This operator returns The ARN of the model created in Amazon SageMaker
:param config: The configuration necessary to create a model.
For details of the configuration parameter see :py:meth:`SageMaker.Client.create_model`
:type config: dict
:param aws_conn_id: The AWS connection ID to use.
:type aws_conn_id: str
.. method:: expand_role(self)
.. method:: execute(self, context)