Fixes to make figures and example script compatible with Neo 0.5.2

code/data_overview_1.py

@@ -534,7 +534,7 @@ el_raw_sig = trial_seg_raw.analogsignals[0]
 
 # plotting raw signal trace
 ax4.plot(el_raw_sig.times.rescale(plotting_time_unit),
-         el_raw_sig.rescale(raw_signal_unit),
+         el_raw_sig.squeeze().rescale(raw_signal_unit),
          color='k')
 
 # setting layout of raw signal plot

code/example.py

@@ -203,7 +203,7 @@ nsx_colors = ['b', 'k', 'r']
 for i, anasig in enumerate(trial_segment.analogsignals):
         plt.plot(
             anasig.times.rescale(time_unit),
-            anasig.rescale(amplitude_unit),
+            anasig.squeeze().rescale(amplitude_unit),
             label=anasig.name,
             color=nsx_colors[i])