Merge pull request #17382 from jklymak/fix-properly-check-for-figrue

QuLogic · web-flow · commit 7377fafe87b4 · 2020-05-13T00:42:02.000-04:00
FIX: properly check figure on gridspec
diff --git a/lib/matplotlib/gridspec.py b/lib/matplotlib/gridspec.py
@@ -293,7 +293,7 @@ def subplots(self, *, sharex=False, sharey=False, squeeze=True,
         .pyplot.subplot
         """
 
-        figure = self[0, 0].get_topmost_subplotspec().get_gridspec().figure
+        figure = self.figure
 
         if figure is None:
             raise ValueError("GridSpec.subplots() only works for GridSpecs "
@@ -555,6 +555,7 @@ def __init__(self, nrows, ncols,
         self._wspace = wspace
         self._hspace = hspace
         self._subplot_spec = subplot_spec
+        self.figure = self._subplot_spec.get_gridspec().figure
         GridSpecBase.__init__(self, nrows, ncols,
                               width_ratios=width_ratios,
                               height_ratios=height_ratios)
