Skip to content

Commit dad08c4

Browse files
committed
fix formatting
1 parent 0eaf41b commit dad08c4

File tree

1 file changed

+32
-32
lines changed

1 file changed

+32
-32
lines changed

src/sagemaker/workflow/_utils.py

Lines changed: 32 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -236,38 +236,38 @@ def __init__(
236236
):
237237
"""Constructor of a register model step.
238238
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`.
271271
"""
272272
super(_RegisterModelStep, self).__init__(name, StepTypeEnum.REGISTER_MODEL, depends_on)
273273
self.estimator = estimator

0 commit comments

Comments
 (0)