@@ -116,5 +116,5 @@ def custom_return(config_path=None, **kwargs):
     )
     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"
     )
Original file line number	Diff line number	Diff line change
`@@ -116,5 +116,5 @@ def custom_return(config_path=None, **kwargs):`
`116`	`116`	`)`
`117`	`117`	`assert (`
`118`	`118`	`image_uri`
`119`		`- == "053634841547.dkr.ecr.us-west-1.amazonaws.com/sagemaker-distribution-prod:3.0.0-gpu"`
	`119`	`+ == "053634841547.dkr.ecr.us-west-1.amazonaws.com/sagemaker-distribution-prod:3.2.0-gpu"`
`120`	`120`	`)`