fix: bump image uri assertion in test_image_retriever

Syed Jafri · Syed Jafri · commit 30ad9d1baee2 · 2026-05-06T04:49:23.000Z
diff --git a/sagemaker-core/tests/integ/image_retriever/test_image_retriever.py b/sagemaker-core/tests/integ/image_retriever/test_image_retriever.py
@@ -27,7 +27,7 @@ def test_retrieve_image_uri():
     )
     assert (
         image_uri
-        == "053634841547.dkr.ecr.us-west-1.amazonaws.com/sagemaker-distribution-prod:3.0.0-gpu"
+        == "053634841547.dkr.ecr.us-west-1.amazonaws.com/sagemaker-distribution-prod:3.2.0-gpu"
     )
 
     image_uri = ImageRetriever.retrieve(

Original file line number	Diff line number	Diff line change
`@@ -27,7 +27,7 @@ def test_retrieve_image_uri():`
`27`	`27`	`)`
`28`	`28`	`assert (`
`29`	`29`	`image_uri`
`30`		`- == "053634841547.dkr.ecr.us-west-1.amazonaws.com/sagemaker-distribution-prod:3.0.0-gpu"`
	`30`	`+ == "053634841547.dkr.ecr.us-west-1.amazonaws.com/sagemaker-distribution-prod:3.2.0-gpu"`
`31`	`31`	`)`
`32`	`32`
`33`	`33`	`image_uri = ImageRetriever.retrieve(`