From b11f7b6f7df92ff210246c5ca1928bc7c99bfc98 Mon Sep 17 00:00:00 2001
From: Anna <anna@mosaicml.com>
Date: Fri, 1 Dec 2023 13:09:19 -0800
Subject: [PATCH] Add more info when run doesnt complete (#2751)

---
 .github/mcli/mcli_pytest.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/.github/mcli/mcli_pytest.py b/.github/mcli/mcli_pytest.py
index e086ef1d2d..21f0ab7fb0 100644
--- a/.github/mcli/mcli_pytest.py
+++ b/.github/mcli/mcli_pytest.py
@@ -102,7 +102,7 @@
         print(line, end='')
 
     print('[GHA] Run completed. Waiting for run to finish...')
-    run = wait_for_run_status(run, status='completed')
+    run = wait_for_run_status(run, status=RunStatus.COMPLETED)
 
-    # Fail if command exited with non-zero exit code or timed out
-    assert run.status == RunStatus.COMPLETED
+    # Fail if command exited with non-zero exit code or timed out (didn't reach COMPLETED)
+    assert run.status == RunStatus.COMPLETED, f'Run {run.name} did not complete: {run.status} ({run.reason})'