Skip to content

Commit 49594ae

Browse files
committed
added exception check
1 parent 39cb429 commit 49594ae

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/sagemaker/image_uris.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -184,10 +184,10 @@ def retrieve(
184184
if config.get("version_aliases").get(original_version):
185185
_version = config.get("version_aliases")[original_version]
186186
if (
187-
config.get("versions")
188-
.get(_version)
189-
.get("version_aliases")
190-
.get(base_framework_version)
187+
config.get("versions", {})
188+
.get(_version, {})
189+
.get("version_aliases", {})
190+
.get(base_framework_version, {})
191191
):
192192
_base_framework_version = config.get("versions")[_version]["version_aliases"][
193193
base_framework_version

0 commit comments

Comments
 (0)