From 70f4b3ef9dcc23120fa70fb5314197aa914270b1 Mon Sep 17 00:00:00 2001 From: Florian Fischer Date: Wed, 4 Sep 2019 23:21:11 +0200 Subject: fix barplot for benchmarks with empty args dict --- src/benchmark.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) (limited to 'src/benchmark.py') diff --git a/src/benchmark.py b/src/benchmark.py index a31552b..c5e96c9 100644 --- a/src/benchmark.py +++ b/src/benchmark.py @@ -531,8 +531,14 @@ class Benchmark: allocators = self.results["allocators"] nallocators = len(allocators) - arg = arg or list(args.keys())[0] - narg = len(args[arg]) + if arg: + arg = args[arg] + elif args.keys(): + arg = args[list(args.keys())[0]] + else: + arg = [""] + + narg = len(arg) for i, allocator in enumerate(allocators): x_vals = list(range(i, narg * (nallocators+1), nallocators+1)) @@ -559,7 +565,7 @@ class Benchmark: color=allocators[allocator]["color"]) plt.legend(loc="best") - plt.xticks(list(range(int(np.floor(nallocators/2)), narg*(nallocators+1), nallocators+1)), args[arg]) + plt.xticks(list(range(int(np.floor(nallocators/2)), narg*(nallocators+1), nallocators+1)), arg) plt.xlabel(eval(xlabel)) plt.ylabel(eval(ylabel)) plt.title(eval(title)) -- cgit v1.2.3