image build fixes

skypilot-org · Sep 15, 2023 · 9c4e338 · 9c4e338
1 parent 657cd6f
commit 9c4e338
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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