diff --git a/poleval/datahandler.py b/poleval/datahandler.py index 791d8f3d64a396e2200f1356b300b8a6c5fd18c8..673b3c53e699d41ceda750f0789e14d6f4753b79 100644 --- a/poleval/datahandler.py +++ b/poleval/datahandler.py @@ -964,7 +964,8 @@ class Plotter: axes_supplied = (axes is not None) if not axes_supplied: fig, axes = plt.subplots(num_rows, num_columns, figsize=(12, 4 * num_rows)) - axes = axes.flatten() + if type(axes) is list: + axes = axes.flatten() # Remove any unused subplots for i in range(num_regions, len(axes)): fig.delaxes(axes[i])