Improve on the query script

var/ramble/repos/builtin/workflow_managers/slurm/workflow_manager.py

@@ -12,7 +12,7 @@
 
 _ensure_job_id_snippet = r"""
 job_id=$(< {experiment_run_dir}/.slurm_job)
-if [ -z "${job_id:-}"]; then
+if [ -z "${job_id:-}" ]; then
     echo "No valid job_id found" 1>&2
     exit 1
 fi
@@ -133,7 +133,13 @@ def __init__(self, dry_run=False):
         )
 
     def generate_query_command(self, job_id):
-        return f"squeue -j {job_id}"
+        return rf"""
+status=$(squeue -h -o "%t" -j {job_id})
+if [ -z "$status" ]; then
+    status=$(sacct -j {job_id} -o state -X -n)
+fi
+echo $status
+        """
 
     def generate_cancel_command(self, job_id):
         return f"scancel {job_id}"