chore: slight adjustment to image uri variant logic

evakravi · evakravi · commit 71ee0ad0c54e · 2023-09-18T20:23:17.000Z
diff --git a/src/sagemaker/jumpstart/types.py b/src/sagemaker/jumpstart/types.py
@@ -373,8 +373,8 @@ def get_image_uri(self, instance_type: str, region: str) -> Optional[str]:
                 .get("image_uri")
             )
 
-        if image_uri_alias is None:
-            return image_uri_alias
+        if image_uri_alias is None or len(image_uri_alias) == 0:
+            return None
 
         if not image_uri_alias.startswith("$"):
             # No leading '$' indicates bad metadata.

Original file line number	Diff line number	Diff line change
`@@ -373,8 +373,8 @@ def get_image_uri(self, instance_type: str, region: str) -> Optional[str]:`
`373`	`373`	`.get("image_uri")`
`374`	`374`	`)`
`375`	`375`
`376`		`- if image_uri_alias is None:`
`377`		`- return image_uri_alias`
	`376`	`+ if image_uri_alias is None or len(image_uri_alias) == 0:`
	`377`	`+ return None`
`378`	`378`
`379`	`379`	`if not image_uri_alias.startswith("$"):`
`380`	`380`	`# No leading '$' indicates bad metadata.`