Skip to content

Commit daf42d2

Browse files
author
Basil Beirouti
committed
formatting fixes
1 parent 9517334 commit daf42d2

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
@@ -2801,7 +2801,7 @@ def create_model_package_from_containers(
28012801
description=None,
28022802
drift_check_baselines=None,
28032803
customer_metadata_properties=None,
2804-
validation_specification=None
2804+
validation_specification=None,
28052805
):
28062806
"""Get request dictionary for CreateModelPackage API.
28072807
@@ -2847,7 +2847,7 @@ def create_model_package_from_containers(
28472847
description,
28482848
drift_check_baselines=drift_check_baselines,
28492849
customer_metadata_properties=customer_metadata_properties,
2850-
validation_specification=validation_specification
2850+
validation_specification=validation_specification,
28512851
)
28522852
if model_package_group_name is not None:
28532853
try:
@@ -4208,7 +4208,7 @@ def get_model_package_args(
42084208
container_def_list=None,
42094209
drift_check_baselines=None,
42104210
customer_metadata_properties=None,
4211-
validation_specification=None
4211+
validation_specification=None,
42124212
):
42134213
"""Get arguments for create_model_package method.
42144214
@@ -4299,7 +4299,7 @@ def get_create_model_package_request(
42994299
tags=None,
43004300
drift_check_baselines=None,
43014301
customer_metadata_properties=None,
4302-
validation_specification=None
4302+
validation_specification=None,
43034303
):
43044304
"""Get request dictionary for CreateModelPackage API.
43054305

0 commit comments

Comments
 (0)