From 374eba3ee22f5ee3866420afbe31d8a71a47bc9a Mon Sep 17 00:00:00 2001 From: Cedric Nugteren Date: Thu, 13 Oct 2022 22:24:35 +0200 Subject: Fix plotting issue with a single row or column --- scripts/benchmark/plot.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/scripts/benchmark/plot.py b/scripts/benchmark/plot.py index c36f476f..2cf5d04d 100644 --- a/scripts/benchmark/plot.py +++ b/scripts/benchmark/plot.py @@ -76,7 +76,12 @@ def plot_graphs(results, file_name, num_rows, num_cols, for col in range(num_cols): index = row * num_cols + col result = results[index] - ax = axes[row, col] + if num_rows == 1: + ax = axes[col] + elif num_cols == 1: + ax = axes[row] + else: + ax = axes[row, col] plt.sca(ax) print("[plot] Plotting subplot %d" % index) -- cgit v1.2.3