Small cleanup

jklymak · jklymak · commit 3dbd7037ace4 · 2019-07-28T14:56:03.000-07:00
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -1840,13 +1840,10 @@ def legend(self, *args, outside=False, ax=None, **kwargs):
             self.legends.append(l)
         else:
             loc = kwargs.pop('loc')
-            if ax is None:
-                gs = self.get_gridspecs()[0]
+            if isinstance(ax, GridSpecBase):
+                gs = ax
             else:
-                if isinstance(ax, GridSpecBase):
-                    gs = ax
-                else:
-                    gs = ax[0].get_gridspec()
+                gs = ax[0].get_gridspec()
             l = gs.legend_outside(loc=loc, align=outside, handles=handles,
                                   labels=labels, **kwargs)
         l._remove_method = self.legends.remove
diff --git a/lib/matplotlib/gridspec.py b/lib/matplotlib/gridspec.py
@@ -181,7 +181,8 @@ def legend_outside(self, handles=None, labels=None, axs=None,
         """
         legend for this gridspec, offset from all the subplots.
 
-        See `.Figure.legend_outside` for details on how to call.
+        See the *outside* argument for `.Figure.legend` for details on how to
+        call.
         """
         if not (self.figure and self.figure.get_constrained_layout()):
             cbook._warn_external('legend_outside method needs '
