diff --git a/python/ray/util/state/state_cli.py b/python/ray/util/state/state_cli.py index 658d841ed17c..b76658775976 100644 --- a/python/ray/util/state/state_cli.py +++ b/python/ray/util/state/state_cli.py @@ -192,6 +192,13 @@ class A(StateSchema): for col in cols: if col in keys: headers.append(col.upper()) + + # --- Fix UX Issue #30805: Prepend TASK_ID --- + if "TASK_ID" in headers: + headers.remove("TASK_ID") + headers.insert(0, "TASK_ID") + # --------------------------------------------- + table.append([data[header.lower()] for header in headers]) return f""" {header}