Amazon
Amazon »

SageMakerRegisterModelVersionOperator

View on GitHub

Created by

Airflow Community

Last Updated

Version

9.10.0

Register a SageMaker model by creating a model version that specifies the model group to which it belongs.

Parameters

image_uri

strREQUIRED

The Amazon EC2 Container Registry (Amazon ECR) path where inference code is stored.

model_url

strREQUIRED

The Amazon S3 path where the model artifacts (the trained weights of the model), which result from model training, are stored. This path must point to a single gzip compressed tar archive (.tar.gz suffix).

package_group_name

strREQUIRED

The name of the model package group that the model is going to be registered to. Will be created if it doesn’t already exist.

The configuration necessary to start a training job (templated)

package_group_desc

str

Description of the model package group, if it was to be created (optional).

package_desc

str

Description of the model package (optional).

model_approval

ApprovalStatus

Approval status of the model package. Defaults to PendingManualApproval

extras

dict | None

Can contain extra parameters for the boto call to create_model_package, and/or overrides for any parameter defined above. For a complete list of available parameters, see https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.create_model_package

The Airflow connection used for AWS credentials. If this is None or empty then the default boto3 behaviour is used. If running Airflow in a distributed manner and aws_conn_id is None or empty, then default boto3 configuration would be used (and must be maintained on each worker node).

AWS region_name. If not specified then the default boto3 behaviour is used.

verify

bool | str | None
Inherited from AwsBaseOperator »

Whether or not to verify SSL certificates. See: https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html

botocore_config

dict | None
Inherited from AwsBaseOperator »

Configuration dictionary (key-values) for botocore client. See: https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html

region

str | None | ArgNotSet
Inherited from AwsBaseOperator »

Documentation

Register a SageMaker model by creating a model version that specifies the model group to which it belongs.

Will create the model group if it does not exist already.

See also

For more information on how to use this operator, take a look at the guide: Register a Sagemaker Model Version