Skip to content

Commit 70da482

Browse files
author
Nikhil Kulkarni
committed
fix black-check
1 parent af73379 commit 70da482

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/sagemaker/model.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -280,9 +280,10 @@ def _compilation_image_uri(self, region, target_instance_type, framework, framew
280280
if framework == "xgboost":
281281
framework_suffix = "-neo"
282282
else:
283-
framework_prefix = "inferentia-" if \
284-
target_instance_type.startswith("ml_inf") else "neo-"
285-
283+
if target_instance_type.startswith("ml_inf"):
284+
framework_prefix = "inferentia"
285+
else:
286+
framework_prefix = "neo-"
286287

287288
return image_uris.retrieve(
288289
"{}{}{}".format(framework_prefix, framework, framework_suffix),

0 commit comments

Comments
 (0)