Skip to content

Commit 7386b67

Browse files
author
Basil Beirouti
committed
formatting fixes
1 parent 0c31aa7 commit 7386b67

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

src/sagemaker/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -305,7 +305,7 @@ def register(
305305
description=None,
306306
drift_check_baselines=None,
307307
customer_metadata_properties=None,
308-
validation_specification=None
308+
validation_specification=None,
309309
):
310310
"""Creates a model package for creating SageMaker models or listing on Marketplace.
311311
@@ -361,7 +361,7 @@ def register(
361361
container_def_list=[container_def],
362362
drift_check_baselines=drift_check_baselines,
363363
customer_metadata_properties=customer_metadata_properties,
364-
validation_specification=validation_specification
364+
validation_specification=validation_specification,
365365
)
366366
model_package = self.sagemaker_session.create_model_package_from_containers(
367367
**model_pkg_args

src/sagemaker/session.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2787,7 +2787,7 @@ def create_model_package_from_containers(
27872787
description=None,
27882788
drift_check_baselines=None,
27892789
customer_metadata_properties=None,
2790-
validation_specification=None
2790+
validation_specification=None,
27912791
):
27922792
"""Get request dictionary for CreateModelPackage API.
27932793
@@ -2833,7 +2833,7 @@ def create_model_package_from_containers(
28332833
description,
28342834
drift_check_baselines=drift_check_baselines,
28352835
customer_metadata_properties=customer_metadata_properties,
2836-
validation_specification=validation_specification
2836+
validation_specification=validation_specification,
28372837
)
28382838
if model_package_group_name is not None:
28392839
try:
@@ -4182,7 +4182,7 @@ def get_model_package_args(
41824182
container_def_list=None,
41834183
drift_check_baselines=None,
41844184
customer_metadata_properties=None,
4185-
validation_specification=None
4185+
validation_specification=None,
41864186
):
41874187
"""Get arguments for create_model_package method.
41884188
@@ -4273,7 +4273,7 @@ def get_create_model_package_request(
42734273
tags=None,
42744274
drift_check_baselines=None,
42754275
customer_metadata_properties=None,
4276-
validation_specification=None
4276+
validation_specification=None,
42774277
):
42784278
"""Get request dictionary for CreateModelPackage API.
42794279

0 commit comments

Comments
 (0)