Show correct status in short dash after early stop

milabench/log.py

@@ -203,6 +203,7 @@ def __init__(self):
  self.live = Live(self.panel, refresh_per_second=4, console=self.console)
  self.rows = defaultdict(dict)
  self.endtimes = {}
+ self.early_stop = {}
 
  def prune(self):
  now = time.time()
@@ -231,6 +232,9 @@ def __call__(self, entry):
  if method:
  method(entry, data, row)
 
+ def on_stop(self, entry, data, row):
+ self.early_stop[entry.tag] = True
+
  def on_end(self, entry, data, row):
  self.endtimes[entry.tag] = time.time()
 
@@ -305,7 +309,7 @@ def on_error(self, entry, data, row):
  def on_end(self, entry, data, row):
  super().on_end(entry, data, row)
  rc = data["return_code"]
- if rc == 0:
+ if rc == 0 or self.early_stop.get(entry.tag, False):
  row["status"] = Text("COMPLETED", style="bold green")
  else:
  row["status"] = Text(f"FAIL:{rc}", style="bold red")