fix

skypilot-org · Apr 29, 2024 · 28358b2 · 28358b2
1 parent 29e6baa
commit 28358b2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_smoke.py b/tests/test_smoke.py
@@ -2727,9 +2727,9 @@ def test_spot_tpu():
         [
             f'sky job launch -n {name} examples/tpu/tpuvm_mnist.yaml -y -d',
             'sleep 5',
-            f'{_SPOT_QUEUE_WAIT}| grep {name} | head -n1 | grep STARTING',
+            f'{_JOB_QUEUE_WAIT}| grep {name} | head -n1 | grep STARTING',
             'sleep 900',  # TPU takes a while to launch
-            f'{_SPOT_QUEUE_WAIT}| grep {name} | head -n1 | grep "RUNNING\|SUCCEEDED"',
+            f'{_JOB_QUEUE_WAIT}| grep {name} | head -n1 | grep "RUNNING\|SUCCEEDED"',
         ],
         _JOB_CANCEL_WAIT.format(job_name=name),
         # Increase timeout since sky job queue -r can be blocked by other spot tests.