Inline _grab_next_args.

anntzer · anntzer · commit a1d79d26b694 · 2018-05-14T23:56:01.000-07:00
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -1109,7 +1109,8 @@ def eventplot(self, positions, orientation='horizontal', lineoffsets=1,
 
         return colls
 
-    # ### Basic plotting
+    #### Basic plotting
+
     # The label_naming happens in `matplotlib.axes._base._plot_args`
     @_preprocess_data(replace_names=["x", "y"],
                       positional_parameter_names=_plot_args_replacer,
@@ -1342,14 +1343,10 @@ def plot(self, *args, scalex=True, scaley=True, **kwargs):
             'k^:'  # black triangle_up markers connected by a dotted line
 
         """
-        lines = []
-
         kwargs = cbook.normalize_kwargs(kwargs, mlines.Line2D._alias_map)
-
-        for line in self._get_lines(*args, **kwargs):
+        lines = [*self._get_lines(*args, **kwargs)]
+        for line in lines:
             self.add_line(line)
-            lines.append(line)
-
         self.autoscale_view(scalex=scalex, scaley=scaley)
         return lines
 
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -169,23 +169,23 @@ def set_prop_cycle(self, *args, **kwargs):
     def __call__(self, *args, **kwargs):
         if self.axes.xaxis is not None and self.axes.yaxis is not None:
             xunits = kwargs.pop('xunits', self.axes.xaxis.units)
-
             if self.axes.name == 'polar':
                 xunits = kwargs.pop('thetaunits', xunits)
+            if xunits != self.axes.xaxis.units:
+                self.axes.xaxis.set_units(xunits)
 
             yunits = kwargs.pop('yunits', self.axes.yaxis.units)
-
             if self.axes.name == 'polar':
                 yunits = kwargs.pop('runits', yunits)
-
-            if xunits != self.axes.xaxis.units:
-                self.axes.xaxis.set_units(xunits)
-
             if yunits != self.axes.yaxis.units:
                 self.axes.yaxis.set_units(yunits)
 
-        ret = self._grab_next_args(*args, **kwargs)
-        return ret
+        while args:
+            this, args = args[:2], args[2:]
+            if args and isinstance(args[0], str):
+                this += args[0],
+                args = args[1:]
+            yield from self._plot_args(this, kwargs)
 
     def get_next_color(self):
         """Return the next color in the cycle."""
@@ -386,14 +386,6 @@ def _plot_args(self, tup, kwargs):
             ret.append(seg)
         return ret
 
-    def _grab_next_args(self, *args, **kwargs):
-        while args:
-            this, args = args[:2], args[2:]
-            if args and isinstance(args[0], str):
-                this += args[0],
-                args = args[1:]
-            yield from self._plot_args(this, kwargs)
-
 
 class _AxesBase(martist.Artist):
     """
