diff options
| -rw-r--r-- | src/benchmark.py | 12 |
1 files changed, 9 insertions, 3 deletions
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)) |