sagemaker.serve.utils.model_package_utils

Utilities for Restricted Model Package support.

Functions

get_s3_uri_from_inference_spec(...)	Extract s3_uri from the first container's model_data_source.
is_restricted_model_package(model_package)	Detect if a model package is a Restricted Model Package.

sagemaker.serve.utils.model_package_utils.get_s3_uri_from_inference_spec(inference_specification) → str | None

Extract s3_uri from the first container’s model_data_source.

Parameters:

inference_specification – The inference_specification from a ModelPackage.

Returns:

The s3_uri string, or None if not available.

sagemaker.serve.utils.model_package_utils.is_restricted_model_package(model_package) → bool

Detect if a model package is a Restricted Model Package.

Parameters:

model_package – A ModelPackage resource object.

Returns:

True if the model package is restricted, False otherwise.