Skip to content

Commit 2fe9cba

Browse files
authored
Update model.py
fixed lines too long
1 parent 5e5c5a1 commit 2fe9cba

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

src/sagemaker/huggingface/model.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,8 @@ def _validate_pt_tf_versions(pytorch_version, tensorflow_version, image_uri):
8585

8686

8787
class HuggingFaceModel(FrameworkModel):
88-
"""A Hugging Face SageMaker ``Model`` that can be deployed to an Amazon SageMaker ``Endpoint``."""
88+
"""A Hugging Face SageMaker ``Model`` that can be deployed
89+
to an Amazon SageMaker ``Endpoint``."""
8990

9091
_framework_name = "huggingface"
9192

@@ -197,14 +198,16 @@ def register(
197198
transform_instances (list): A list of the instance types on which a transformation
198199
job can be run or on which an endpoint can be deployed.
199200
model_package_name (str): Model Package name, exclusive to `model_package_group_name`,
200-
using `model_package_name` makes the Model Package un-versioned. Defaults to ``None``.
201+
using `model_package_name` makes the Model Package un-versioned.
202+
Defaults to ``None``.
201203
model_package_group_name (str): Model Package Group name, exclusive to
202204
`model_package_name`, using `model_package_group_name` makes the Model Package
203205
versioned. Defaults to ``None``.
204206
image_uri (str): Inference image URI for the container. Model class' self.image will
205207
be used if it is None. Defaults to ``None``.
206208
model_metrics (ModelMetrics): ModelMetrics object. Defaults to ``None``.
207-
metadata_properties (MetadataProperties): MetadataProperties object. Defaults to ``None``.
209+
metadata_properties (MetadataProperties): MetadataProperties object.
210+
Defaults to ``None``.
208211
marketplace_cert (bool): A boolean value indicating if the Model Package is certified
209212
for AWS Marketplace. Defaults to ``False``.
210213
approval_status (str): Model Approval Status, values can be "Approved", "Rejected",

0 commit comments

Comments
 (0)