From c1a44664bc6c668e01963824b79b57c025f158dd Mon Sep 17 00:00:00 2001 From: David Stansby Date: Mon, 8 Apr 2019 13:36:15 +0100 Subject: [PATCH] Allow plot_input to pass through kwargs --- pfsspy/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pfsspy/__init__.py b/pfsspy/__init__.py index 435f20e6..2381f6fe 100644 --- a/pfsspy/__init__.py +++ b/pfsspy/__init__.py @@ -144,7 +144,7 @@ def __init__(self, br, nr, rss, dtime=None): nphi = self.br.shape[1] self.grid = Grid(ns, nphi, nr, rss) - def plot_input(self, ax=None): + def plot_input(self, ax=None, **kwargs): """ Plot a 2D image of the magnetic field boundary condition. @@ -153,7 +153,7 @@ def plot_input(self, ax=None): ax : Axes Axes to plot to. If ``None``, creates a new figure. """ - mesh = pfsspy.plot.radial_cut(self.grid.pc, self.grid.sc, self.br, ax) + mesh = pfsspy.plot.radial_cut(self.grid.pc, self.grid.sc, self.br, ax, **kwargs) return mesh