File tree 3 files changed +5
-3
lines changed
3 files changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -359,7 +359,7 @@ def register(
359
359
description = description ,
360
360
container_def_list = [container_def ],
361
361
drift_check_baselines = drift_check_baselines ,
362
- customer_metadata_properties = customer_metadata_properties ,
362
+ customer_metadata_properties = customer_metadata_properties ,
363
363
)
364
364
model_package = self .sagemaker_session .create_model_package_from_containers (
365
365
** model_pkg_args
Original file line number Diff line number Diff line change @@ -2828,7 +2828,7 @@ def create_model_package_from_containers(
2828
2828
try :
2829
2829
self .sagemaker_client .describe_model_package_group (
2830
2830
ModelPackageGroupName = request ["ModelPackageGroupName" ])
2831
- except ClientError as e :
2831
+ except ClientError :
2832
2832
self .sagemaker_client .create_model_package_group (
2833
2833
ModelPackageGroupName = request ["ModelPackageGroupName" ])
2834
2834
return self .sagemaker_client .create_model_package (** request )
Original file line number Diff line number Diff line change @@ -230,6 +230,7 @@ def test_register_model_package(
230
230
assert result is not None
231
231
sagemaker_session .sagemaker_client .delete_model_package (ModelPackageName = model_package_name )
232
232
233
+
233
234
def test_register_model_package_via_group (
234
235
mxnet_training_job ,
235
236
sagemaker_session ,
@@ -266,12 +267,13 @@ def test_register_model_package_via_group(
266
267
result = predictor .predict (data )
267
268
assert result is not None
268
269
model_packages = \
269
- sagemaker_session .sagemaker_client .list_model_packages (ModelPackageGroupName = model_package_group_name )[
270
+ sagemaker_session .sagemaker_client .list_model_packages (ModelPackageGroupName = model_package_group_name )[
270
271
'ModelPackageSummaryList' ]
271
272
for model_package in model_packages :
272
273
sagemaker_session .sagemaker_client .delete_model_package (ModelPackageName = model_package ['ModelPackageArn' ])
273
274
sagemaker_session .sagemaker_client .delete_model_package_group (ModelPackageGroupName = model_package_group_name )
274
275
276
+
275
277
def test_register_model_package_versioned (
276
278
mxnet_training_job ,
277
279
sagemaker_session ,
You can’t perform that action at this time.
0 commit comments