Skip to content

Commit 3b4cf96

Browse files
author
Payton Staub
committed
Pylint, black-format
1 parent 3d4f67f commit 3b4cf96

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

src/sagemaker/workflow/step_collections.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -96,8 +96,9 @@ def __init__(
9696
specified, a compiled model is used (default: None).
9797
description (str): Model Package description (default: None).
9898
tags (List[dict[str, str]]): The list of tags to attach to the model package group. Note
99-
that tags will only be applied to newly created model package groups; if the
100-
name of an existing group is passed to "model_package_group_name", tags will not be applied.
99+
that tags will only be applied to newly created model package groups; if the
100+
name of an existing group is passed to "model_package_group_name",
101+
tags will not be applied.
101102
**kwargs: additional arguments to `create_model`.
102103
"""
103104
steps: List[Step] = []

tests/unit/sagemaker/workflow/test_step_collections.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ def test_register_model(estimator, model_metrics):
182182
approval_status="Approved",
183183
description="description",
184184
depends_on=["TestStep"],
185-
tags=[{"Key": "myKey", "Value": "myValue"}]
185+
tags=[{"Key": "myKey", "Value": "myValue"}],
186186
)
187187
assert ordered(register_model.request_dicts()) == ordered(
188188
[
@@ -211,7 +211,7 @@ def test_register_model(estimator, model_metrics):
211211
},
212212
"ModelPackageDescription": "description",
213213
"ModelPackageGroupName": "mpg",
214-
"Tags": [{"Key": "myKey", "Value": "myValue"}]
214+
"Tags": [{"Key": "myKey", "Value": "myValue"}],
215215
},
216216
},
217217
]

0 commit comments

Comments
 (0)