Skip to content

Commit af73379

Browse files
author
Nikhil Kulkarni
committed
fix pylint
1 parent 7ff4d55 commit af73379

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/sagemaker/model.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -276,11 +276,12 @@ def _compilation_image_uri(self, region, target_instance_type, framework, framew
276276
"""
277277
framework_prefix = ""
278278
framework_suffix = ""
279-
279+
280280
if framework == "xgboost":
281281
framework_suffix = "-neo"
282282
else:
283-
framework_prefix = "inferentia-" if target_instance_type.startswith("ml_inf") else "neo-"
283+
framework_prefix = "inferentia-" if \
284+
target_instance_type.startswith("ml_inf") else "neo-"
284285

285286

286287
return image_uris.retrieve(

0 commit comments

Comments
 (0)