Merge pull request #420 from ruteee/master

Reversion of PR #415
BindsNET · Oct 6, 2020 · e69b975 · e69b975
2 parents ff5925d + 6917d98
commit e69b975
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)