@@ -236,38 +236,38 @@ def __init__(
236
236
):
237
237
"""Constructor of a register model step.
238
238
239
- Args:
240
- name (str): The name of the training step.
241
- step_type (StepTypeEnum): The type of the step with value
242
- `StepTypeEnum.Training`.
243
- estimator (EstimatorBase): A `sagemaker.estimator.EstimatorBase` instance.
244
- model_data: the S3 URI to the model data from training.
245
- content_types (list): The supported MIME types for the
246
- input data (default: None).
247
- response_types (list): The supported MIME types for
248
- the output data (default: None).
249
- inference_instances (list): A list of the instance types that are used to
250
- generate inferences in real-time (default: None).
251
- transform_instances (list): A list of the instance types on which a
252
- transformation job can be run or on which an endpoint
253
- can be deployed (default: None).
254
- model_package_group_name (str): Model Package Group name, exclusive to
255
- `model_package_name`, using `model_package_group_name`
256
- makes the Model Package versioned (default: None).
257
- model_metrics (ModelMetrics): ModelMetrics object (default: None).
258
- metadata_properties (MetadataProperties): MetadataProperties object
259
- (default: None).
260
- approval_status (str): Model Approval Status, values can be "Approved",
261
- "Rejected", or "PendingManualApproval"
262
- (default: "PendingManualApproval").
263
- image_uri (str): The container image uri for Model Package, if not specified,
264
- Estimator's training container image will be used (default: None).
265
- compile_model_family (str): Instance family for compiled model,
266
- if specified, a compiled model will be used (default: None).
267
- description (str): Model Package description (default: None).
268
- depends_on (List[str] or List[Step]): A list of step names or instances
269
- this step depends on
270
- **kwargs: additional arguments to `create_model`.
239
+ Args:
240
+ name (str): The name of the training step.
241
+ step_type (StepTypeEnum): The type of the step with value
242
+ `StepTypeEnum.Training`.
243
+ estimator (EstimatorBase): A `sagemaker.estimator.EstimatorBase` instance.
244
+ model_data: the S3 URI to the model data from training.
245
+ content_types (list): The supported MIME types for the
246
+ input data (default: None).
247
+ response_types (list): The supported MIME types for
248
+ the output data (default: None).
249
+ inference_instances (list): A list of the instance types that are used to
250
+ generate inferences in real-time (default: None).
251
+ transform_instances (list): A list of the instance types on which a
252
+ transformation job can be run or on which an endpoint
253
+ can be deployed (default: None).
254
+ model_package_group_name (str): Model Package Group name, exclusive to
255
+ `model_package_name`, using `model_package_group_name`
256
+ makes the Model Package versioned (default: None).
257
+ model_metrics (ModelMetrics): ModelMetrics object (default: None).
258
+ metadata_properties (MetadataProperties): MetadataProperties object
259
+ (default: None).
260
+ approval_status (str): Model Approval Status, values can be "Approved",
261
+ "Rejected", or "PendingManualApproval"
262
+ (default: "PendingManualApproval").
263
+ image_uri (str): The container image uri for Model Package, if not specified,
264
+ Estimator's training container image will be used (default: None).
265
+ compile_model_family (str): Instance family for compiled model,
266
+ if specified, a compiled model will be used (default: None).
267
+ description (str): Model Package description (default: None).
268
+ depends_on (List[str] or List[Step]): A list of step names or instances
269
+ this step depends on
270
+ **kwargs: additional arguments to `create_model`.
271
271
"""
272
272
super (_RegisterModelStep , self ).__init__ (name , StepTypeEnum .REGISTER_MODEL , depends_on )
273
273
self .estimator = estimator
0 commit comments