Skip to content

Commit 3ba08b6

Browse files
authored
Merge pull request #620 from gdsfactory/better_title_for_plot_model
better title for plot model
2 parents f04327a + 47fd4cb commit 3ba08b6

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

gplugins/sax/plot_model.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ def plot_model(
1919
wavelength_stop: float = 1.6,
2020
wavelength_points: int = 2000,
2121
phase: bool = False,
22+
title: str | None = None,
2223
) -> None:
2324
"""Plot Model Sparameters Magnitude.
2425
@@ -32,6 +33,7 @@ def plot_model(
3233
wavelength_stop: wavelength max (um).
3334
wavelength_points: number of wavelength steps.
3435
phase: plot phase instead of magnitude.
36+
title: plot title.
3537
3638
.. plot::
3739
:include-source:
@@ -67,7 +69,8 @@ def plot_model(
6769
y = 20 * np.log10(y) if logscale else y
6870
ylabel = "|S (dB)|" if logscale else "|S|"
6971
ax.plot(wavelengths, y, label=port2)
70-
ax.set_title(port1)
72+
73+
ax.set_title(title or f"{model.__name__} S-Parameters")
7174
ax.set_xlabel("wavelength (um)")
7275
ax.set_ylabel(ylabel)
7376
plt.legend()

0 commit comments

Comments
 (0)