Fix reports

mila-iqia · Apr 2, 2024 · e8d7bea · e8d7bea
1 parent 7996e56
commit e8d7bea
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/milabench/common.py b/milabench/common.py
@@ -355,10 +355,9 @@ def _push_reports(reports_repo, runs, packs:dict=None):
 
         meta = [e["data"] for _r in reports for e in _r if e["event"] == "meta"]
 
-        for _meta in meta:
-            for gpu in _meta["accelerators"]["gpus"].values():
-                device = gpu["product"].replace(" ", "_")
-                break
+        for gpu in (_ for _meta in meta for _ in _meta["accelerators"]["gpus"].values()):
+            device = gpu["product"].replace(" ", "_")
+            break
         else:
             for _meta in meta:
                 device = _meta["cpu"]["brand"].replace(" ", "_")