fix: fix error retrieving status for queued jobs

ai4os · Nov 22, 2024 · c2a3fc6 · c2a3fc6
1 parent 5626029
commit c2a3fc6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ai4papi/nomad/common.py b/ai4papi/nomad/common.py
@@ -213,7 +213,9 @@ def get_deployment(
 
         # Replace Nomad status with a more user-friendly status
         # Final list includes: starting, down, running, complete, failed, ...
-        status = a['TaskStates']['main']['State'] # more relevant than a['ClientStatus']
+        # We use the status of the "main" task because it isn more relevant the the
+        # status of the overall job (a['ClientStatus'])
+        status = a['TaskStates']['main']['State'] if a.get('TaskStates') else 'queued'
         status_map = {  # nomad: papi
             'pending': 'starting',
             'unknown': 'down',