
SageMakerRegisterModelVersionOperator
Created by
Airflow Community
Version
Register a SageMaker model by creating a model version that specifies the model group to which it belongs.
Parameters
image_uri
strREQUIRED
image_uri
strREQUIREDThe Amazon EC2 Container Registry (Amazon ECR) path where inference code is stored.
model_url
strREQUIRED
model_url
strREQUIREDThe 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
package_group_name
strREQUIREDThe 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
package_group_desc
strDescription of the model package group, if it was to be created (optional).
package_desc
str
package_desc
strDescription of the model package (optional).
model_approval
ApprovalStatus
model_approval
ApprovalStatusApproval status of the model package. Defaults to PendingManualApproval
extras
dict | None
extras
dict | NoneCan 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.
Whether or not to verify SSL certificates. See: https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html
Configuration dictionary (key-values) for botocore client. See: https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html
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