Rerversion of PR #415

BindsNET · Oct 6, 2020 · 6917d98 · 6917d98
1 parent ff5925d
commit 6917d98
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)