Rename fig= to figure= in TiledDataset.plot

changelog/491.feature.rst

@@ -1 +1 @@
-Add a ``fig=`` keyword argument to `TiledDataset.plot` and make it default to the current figure.
+Add a ``figure=`` keyword argument to `TiledDataset.plot` and make it default to the current figure.

dkist/dataset/tiled_dataset.py

@@ -167,7 +167,7 @@
                 ylabel = coord.get_axislabel() or coord._get_default_axislabel()
         return (xlabel, ylabel)
 
-    def plot(self, slice_index, share_zscale=False, fig=None, **kwargs):
+    def plot(self, slice_index, share_zscale=False, figure=None, **kwargs):
         """
         Plot a slice of each tile in the TiledDataset
 
@@ -181,32 +181,32 @@
             Determines whether the color scale of the plots should be calculated
             independently (``False``) or shared across all plots (``True``).
             Defaults to False
-        fig : `matplotlib.figure.Figure`
+        figure : `matplotlib.figure.Figure`
             A figure to use for the plot. If not specified the current pyplot
             figure will be used, or a new one created.
         """
         if isinstance(slice_index, int):
             slice_index = (slice_index,)
         vmin, vmax = np.inf, 0
 
-        if fig is None:
-            fig = plt.gcf()
+        if figure is None:
+            figure = plt.gcf()
 
         tiles = self.slice_tiles[slice_index].flat
         for i, tile in enumerate(tiles):
-            ax = fig.add_subplot(self.shape[0], self.shape[1], i+1, projection=tile.wcs)
+            ax = figure.add_subplot(self.shape[0], self.shape[1], i+1, projection=tile.wcs)
             tile.plot(axes=ax, **kwargs)
             if i == 0:
                 xlabel, ylabel = self._get_axislabels(ax)
-                fig.supxlabel(xlabel, y=0.05)
-                fig.supylabel(ylabel, x=0.05)
+                figure.supxlabel(xlabel, y=0.05)
+                figure.supylabel(ylabel, x=0.05)
             axmin, axmax = ax.get_images()[0].get_clim()
             vmin = axmin if axmin < vmin else vmin
             vmax = axmax if axmax > vmax else vmax
             ax.set_ylabel(" ")
             ax.set_xlabel(" ")
         if share_zscale:
-            for ax in fig.get_axes():
+            for ax in figure.get_axes():
                 ax.get_images()[0].set_clim(vmin, vmax)
         title = f"{self.inventory['instrumentName']} Dataset ({self.inventory['datasetId']}) at "
         for i, (coord, val) in enumerate(list(tiles[0].global_coords.items())[::-1]):
@@ -216,8 +216,8 @@
                 val = val.symbol
             title += f"{coord} {val}" + (", " if i != len(slice_index)-1 else " ")
         title += f"(slice={(slice_index if len(slice_index) > 1 else slice_index[0])})".replace("slice(None, None, None)", ":")
-        fig.suptitle(title, y=0.95)
-        return fig
+        figure.suptitle(title, y=0.95)
+        return figure
 
     @property
     def slice_tiles(self):