Merge pull request #90 from dstansby/plotinkwargs

Allow plot_input to pass through kwargs
dstansby · Apr 9, 2019 · 7dbd2b5 · 7dbd2b5
2 parents 59ec616 + c1a4466
commit 7dbd2b5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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