From 9c4e3387878cb38b6ffaee4c75a0980dc316546f Mon Sep 17 00:00:00 2001
From: Romil Bhardwaj <romil.bhardwaj@gmail.com>
Date: Fri, 15 Sep 2023 15:40:42 -0700
Subject: [PATCH] image build fixes

---
 tests/kubernetes/build_image.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/kubernetes/build_image.sh b/tests/kubernetes/build_image.sh
index 84c3715e1e4..675a15b12bc 100755
--- a/tests/kubernetes/build_image.sh
+++ b/tests/kubernetes/build_image.sh
@@ -32,7 +32,7 @@ while getopts ":pg" opt; do
 done
 
 # Add -gpu to the tag if the GPU image is being built
-if [[ $gpu ]]; then
+if [[ $gpu == "true" ]]; then
   TAG=$TAG-gpu:latest
 else
   TAG=$TAG:latest