From 23f2afb2ce8529014210262a3fecf6566362f292 Mon Sep 17 00:00:00 2001
From: michael-amd <Michael.Zhang@amd.com>
Date: Fri, 8 Aug 2025 18:51:29 -0700
Subject: [PATCH] [AMD] Update SGLang image fallback logic for AMD CI (#8980)

---
 scripts/amd_ci_start_container.sh | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/scripts/amd_ci_start_container.sh b/scripts/amd_ci_start_container.sh
index b9c2902e8..ebb41debf 100755
--- a/scripts/amd_ci_start_container.sh
+++ b/scripts/amd_ci_start_container.sh
@@ -135,8 +135,7 @@ else
 fi
 
 # Find and pull the latest image
-IMAGE=$(find_latest_image "${GPU_ARCH}")
-if [ $? -eq 0 ]; then
+if IMAGE=$(find_latest_image "${GPU_ARCH}"); then
   echo "Pulling Docker image: $IMAGE"
 else
   echo "$FALLBACK_MSG" >&2