Add OGB protocol diagnosticsHEADmain

1 files changed, 5 insertions, 1 deletions

diff --git a/rrog/collect_results.py b/rrog/collect_results.py
index 125c0ab..4c1390c 100644
--- a/rrog/collect_results.py
+++ b/rrog/collect_results.py
@@ -84,7 +84,11 @@ def _is_classic_baseline(rep: dict) -> bool:
 
 
 def _compute_label(rep: dict) -> str:
-    label = str(rep["compute"])
+    compute = str(rep["compute"])
+    if _is_classic_baseline(rep):
+        label = "classic"
+    else:
+        label = f"{compute}-T{rep.get('T')}-ns{rep.get('n_sup')}"
     ema = float(rep.get("ema", 0.0) or 0.0)
     if ema > 0:
         label += f"+ema{ema:g}"