Add color_channels option in in plot_probe_map

[RobertoDF]

Fix #2739

[alejoe91]

Thanks @RobertoDF, I think it's a good solution. Can you add a code example here and a screenshot to show how it looks?

[RobertoDF]

Here it is:

# two probe groups in rec
bad_channel_ids, channel_labels = detect_bad_channels(high_passed_rec, dead_channel_threshold=-1)

color_channels = [channel_label_color_dict[label] for label in channel_labels]

si.plot_probe_map(raw_rec, color_channels=color_channels, with_channel_ids=False)

[alejoe91]

@RobertoDF one last suggestion. I think that instead of passing a list of one elements, one easier option is to allow this:

plot_probe_map(rec, color_channels="r")

See implementation details in review :)

[RobertoDF]

definitely better 👍👍

[alejoe91]

@RobertoDF can you fix the conflict? Looks good to merge after that!

[RobertoDF]

ah yes, my mistake. hope it is ok now.