Add step

src/validator_states.py

@@ -46,13 +46,14 @@ def create_validator_states() -> gr.Dataframe:
             len(run.invalid_submissions),
             colorize(f"{timedelta(seconds=int(run.average_benchmarking_time))}", "orange" if run.average_benchmarking_time > AVERAGE_BENCHMARKING_TIME_WARNING_THRESHOLD else "springgreen" if run.average_benchmarking_time > 0 else "gray"),
             colorize(f"{timedelta(seconds=run.eta)}", "orange" if run.eta > ETA_WARNING_THRESHOLD else "springgreen" if run.eta > 0 else "gray"),
+            run.step,
             colorize(f"{vtrust:.4f}", "springgreen" if vtrust > VTRUST_WARNING_THRESHOLD else "red"),
             colorize(updated, "springgreen" if updated < UPDATED_WARNING_THRESHOLD else "red"),
         ])
 
     return gr.Dataframe(
-        pd.DataFrame(data, columns=["UID", "Name", "Version", "Status", "Winner", "Tested", "Invalid", "Avg. Benchmark Time", "ETA", "VTrust", "Updated"]),
-        datatype=["number", "markdown", "markdown", "markdown", "markdown", "number", "number", "markdown", "markdown", "markdown", "markdown"],
+        pd.DataFrame(data, columns=["UID", "Name", "Version", "Status", "Winner", "Tested", "Invalid", "Avg. Benchmark Time", "ETA", "Step", "VTrust", "Updated"]),
+        datatype=["number", "markdown", "markdown", "markdown", "markdown", "number", "number", "markdown", "markdown", "number", "markdown", "markdown"],
         interactive=False,
         max_height=800,
     )

src/wandb_data.py

@@ -71,6 +71,7 @@ class Run:
     hotkey: str
     status: BenchmarkStatus
     average_benchmarking_time: float
+    step: int
     eta: int
     winner_uid: int | None
     baseline_metrics: MetricData | None
@@ -200,6 +201,7 @@ def _add_runs(wandb_runs: list[wapi.Run]):
             hotkey=hotkey,
             status=status,
             average_benchmarking_time=average_benchmarking_time,
+            step=int(metrics["step"]),
             eta=max(int(average_benchmarking_time * (len(submission_info) - len(submissions) - len(invalid_submissions))) if average_benchmarking_time else 0, 0) if status != BenchmarkStatus.FINISHED else 0,
             winner_uid=winner_uid,
             baseline_metrics=baseline_metrics,