Skip to content

Commit 46dd800

Browse files
committed
Fixing lint errors
1 parent 538dbdf commit 46dd800

File tree

3 files changed

+5
-3
lines changed

3 files changed

+5
-3
lines changed

src/sagemaker/model.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -359,7 +359,7 @@ def register(
359359
description=description,
360360
container_def_list=[container_def],
361361
drift_check_baselines=drift_check_baselines,
362-
customer_metadata_properties = customer_metadata_properties,
362+
customer_metadata_properties=customer_metadata_properties,
363363
)
364364
model_package = self.sagemaker_session.create_model_package_from_containers(
365365
**model_pkg_args

src/sagemaker/session.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -2828,7 +2828,7 @@ def create_model_package_from_containers(
28282828
try:
28292829
self.sagemaker_client.describe_model_package_group(
28302830
ModelPackageGroupName=request["ModelPackageGroupName"])
2831-
except ClientError as e:
2831+
except ClientError:
28322832
self.sagemaker_client.create_model_package_group(
28332833
ModelPackageGroupName=request["ModelPackageGroupName"])
28342834
return self.sagemaker_client.create_model_package(**request)

tests/integ/test_mxnet.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -230,6 +230,7 @@ def test_register_model_package(
230230
assert result is not None
231231
sagemaker_session.sagemaker_client.delete_model_package(ModelPackageName=model_package_name)
232232

233+
233234
def test_register_model_package_via_group(
234235
mxnet_training_job,
235236
sagemaker_session,
@@ -266,12 +267,13 @@ def test_register_model_package_via_group(
266267
result = predictor.predict(data)
267268
assert result is not None
268269
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)[
270271
'ModelPackageSummaryList']
271272
for model_package in model_packages:
272273
sagemaker_session.sagemaker_client.delete_model_package(ModelPackageName=model_package['ModelPackageArn'])
273274
sagemaker_session.sagemaker_client.delete_model_package_group(ModelPackageGroupName=model_package_group_name)
274275

276+
275277
def test_register_model_package_versioned(
276278
mxnet_training_job,
277279
sagemaker_session,

0 commit comments

Comments
 (0)