diff --git a/summarize.py b/summarize.py index 0648b052738de913885d93f4e5a78c76dd9ef848..d8b3e04553913c61448bd42da13cbd23c4cea06c 100755 --- a/summarize.py +++ b/summarize.py @@ -46,12 +46,16 @@ def collect(result_dir) -> Data: return data -Avgs = dict[Variant, float] +Avgs = dict[Variant, dict[str, float]] def calc_avgs(data) -> Avgs: """Calculate only averages from data""" - avgs = {t: numpy.mean(ms) for t, ms in data.items()} + avgs = { + t: {k: numpy.mean(ms) + for k, ms in d.items()} + for t, d in data.items() + } return avgs