diff --git a/ngclearn/components/base_monitor.py b/ngclearn/components/base_monitor.py index 00c338b2a..8238264d8 100644 --- a/ngclearn/components/base_monitor.py +++ b/ngclearn/components/base_monitor.py @@ -310,4 +310,4 @@ def make_plot(self, compartment, ax=None, ylabel=None, xlabel=None, title=None, for k in range(n): _ax.plot(vals[:, 0, k]) else: - plot_func(vals, ax=_ax) + plot_func(vals[:, :, 0:n], ax=_ax)