Label of plots in legend

[maestapereira]

I was trying to mix plt plot functions with spiketools plot functions and manually add a legend and noticed the spiketools plot function label did not work properly.

Here are the scenarios:

1. Weird legend:

ax = get_grid_subplot(grid, 1, slice(0, 3))
a1 = ax.plot(trace_times, trace_values, label='trace')
a2 = plot_scatter(spike_times, np.zeros(len(spike_times)), c='r', ax=ax, label='spikes')
ax.legend([a1, a2])

This shows a legend, but the spiketools plot has no label and the plt plot has a weird label (not "trace").

2. No legend entirely:

ax = get_grid_subplot(grid, 1, slice(0, 3))
a1 = ax.plot(trace_times, trace_values)
a2 = plot_scatter(spike_times, np.zeros(len(spike_times)), c='r', ax=ax)
ax.legend([a1, a2], ['trace', 'spikes'])

Which doesn't show a legend (just a tiny square where the legend is supposed to be. Adding the handles= and labels= to the above code in ax.legend also shoes no legend.