Skip to content

Commit edbf61f

Browse files
change InferenceImage lookup in compilation job description API response to get
1 parent 08c3a7b commit edbf61f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/model.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -658,7 +658,7 @@ def compile(
658658
if target_instance_family == "ml_eia2":
659659
pass
660660
elif target_instance_family.startswith("ml_"):
661-
self.image_uri = job_status["InferenceImage"]
661+
self.image_uri = job_status.get("InferenceImage", None)
662662
self._is_compiled_model = True
663663
else:
664664
LOGGER.warning(

0 commit comments

Comments
 (0)