Add ax parameter to yasa.topoplot

[raphaelvallat]

Closes #237

Summary

• Added an ax keyword argument to yasa.topoplot() so the plot can be drawn on a caller-supplied matplotlib.axes.Axes (e.g. a subplot cell).
• When ax=None (default), behaviour is identical to before — a new figure is created using figsize and dpi.
• When an ax is provided, the figure is retrieved via ax.get_figure() and the colorbar is placed with fig.colorbar(..., ax=ax) so it adapts to the subplot layout automatically.

Usage

import yasa
import pandas as pd
import matplotlib.pyplot as plt

data1 = pd.Series([4, 8, 7, 1, 2, 3, 5], index=["F4", "F3", "C4", "C3", "P3", "P4", "Oz"], name="NREM")
data2 = pd.Series([2, 5, 4, 3, 6, 7, 4], index=["F4", "F3", "C4", "C3", "P3", "P4", "Oz"], name="REM")

fig, axes = plt.subplots(1, 2, figsize=(8, 4))
yasa.topoplot(data1, ax=axes[0])
yasa.topoplot(data2, ax=axes[1])

Changes

File	Change
src/yasa/plotting.py	Added ax=None parameter, subplot-aware colorbar logic, and a new docstring example

[raphaelvallat]

This bug has been fixed in the latest version of the Menghini pipeline — I was using an outdated version

[codecov-commenter]

Codecov Report

❌ Patch coverage is 44.44444% with 5 lines in your changes missing coverage. Please review.
✅ Project coverage is 92.41%. Comparing base (2319677) to head (92697a5).

Files with missing lines	Patch %	Lines
src/yasa/plotting.py	44.44%	3 Missing and 2 partials ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##           master     #244      +/-   ##
==========================================
- Coverage   92.57%   92.41%   -0.17%     
==========================================
  Files          13       13              
  Lines        2814     2820       +6     
  Branches      383      385       +2     
==========================================
+ Hits         2605     2606       +1     
- Misses        153      156       +3     
- Partials       56       58       +2     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.