make sure all gridspecs have a figure attribute

jklymak · jklymak · commit d79498b26b04 · 2020-05-11T15:16:42.000-07:00
diff --git a/lib/matplotlib/gridspec.py b/lib/matplotlib/gridspec.py
@@ -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)
