From 6917d98516a13f90e98ef87e79c25c39fa2318ae Mon Sep 17 00:00:00 2001 From: rute Date: Tue, 6 Oct 2020 11:48:44 -0300 Subject: [PATCH] Rerversion of PR #415 --- bindsnet/analysis/plotting.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/bindsnet/analysis/plotting.py b/bindsnet/analysis/plotting.py index 21f89c2c..663b93e3 100644 --- a/bindsnet/analysis/plotting.py +++ b/bindsnet/analysis/plotting.py @@ -98,13 +98,13 @@ def plot_spikes( if time is None: # Set it for entire duration for key in spikes.keys(): - time = (0, spikes[key].shape[1]) + time = (0, spikes[key].shape[0]) break # Use all neurons if no argument provided. for key, val in spikes.items(): if key not in n_neurons.keys(): - n_neurons[key] = (0, val.shape[0]) + n_neurons[key] = (0, val.shape[1]) if ims is None: fig, axes = plt.subplots(n_subplots, 1, figsize=figsize)