diff --git a/milabench/common.py b/milabench/common.py index 92bde8c4e..a9edc8b31 100644 --- 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(" ", "_")