Package

Model packaging step

SageMaker step type

Processing

Node type

internal (consumes upstream, produces downstream)

Container entry point

package.py

Interface file

steps/interfaces/package.step.yaml

Compute

Compute kind

sklearn

Functionality

MIMS packaging script that extracts model artifacts, includes calibration model if available, copies inference scripts, and creates a packaged model.tar.gz for deployment.

Inputs (dependencies)

Input

Type

Required

Compatible producers

model_input

model_artifacts

yes

XGBoostTraining, TrainingStep, ModelStep, PyTorchTraining, XgboostMtTraining, LightGBMMTTraining

inference_scripts_input

custom_property

no

ProcessingStep, ScriptStep

calibration_model

processing_output

no

ModelCalibration, PercentileModelCalibration

Outputs

Output

Type

packaged_model

model_artifacts

Consumers (downstream steps)

Steps that declare this step as a compatible input source:

Framework requirements

Package

Version

python

>=3.7


Back to the Step Catalog