Cleanup matplotlib.use

timhoffm · timhoffm · commit c6e066062687 · 2019-01-06T11:25:20.000+01:00
diff --git a/lib/matplotlib/__init__.py b/lib/matplotlib/__init__.py
@@ -1297,15 +1297,15 @@ def __exit__(self, exc_type, exc_value, exc_tb):
         self.__fallback()
 
 
-def use(arg, warn=False, force=True):
+def use(backend, warn=False, force=True):
     """
-    Set the matplotlib backend to one of the known backends.
+    Select the matplotlib backend for rendering.
 
     Parameters
     ----------
-    arg : str
-        The backend to switch to.  This can either be one of the
-        'standard' backend names:
+    backend : str
+        The backend to switch to.  This can either be one of the standard
+        backend names:
 
         - interactive backends:
           GTK3Agg, GTK3Cairo, MacOSX, nbAgg,
@@ -1319,37 +1319,34 @@ def use(arg, warn=False, force=True):
 
         Note: Standard backend names are case-insensitive here.
 
-    warn : bool, optional
-        If True, warn if this is called after pyplot has been imported
-        and a backend is set up.
+    warn : bool, optional, default: False
+        If True and not *force*, warn that the call will have no effect if
+        this is called after pyplot has been imported and a backend is set up.
 
-        defaults to False.
-
-    force : bool, optional
+    force : bool, optional, default: True
         If True, attempt to switch the backend.   An ImportError is raised if
         an interactive backend is selected, but another interactive
-        backend has already started.  This defaults to True.
+        backend has already started.
 
     See Also
     --------
     :ref:`backends`
     matplotlib.get_backend
     """
-    name = validate_backend(arg)
+    name = validate_backend(backend)
 
-    # if setting back to the same thing, do nothing
-    if (dict.__getitem__(rcParams, 'backend') == name):
+    if dict.__getitem__(rcParams, 'backend') == name:
+        # Nothing to do if the requested backend is already set
         pass
-
-    # Check if we have already imported pyplot and triggered
-    # backend selection, do a bit more work
     elif 'matplotlib.pyplot' in sys.modules:
-        # If we are here then the requested is different than the current.
+        # pyplot has already been imported (which triggered backend selection)
+        # and the requested backend is different from the current one.
+
         # If we are going to force the switch, never warn, else, if warn
         # is True, then direct users to `plt.switch_backend`
         if (not force) and warn:
             cbook._warn_external(
-                ("matplotlib.pyplot as already been imported, "
+                ("matplotlib.pyplot has already been imported, "
                  "this call will have no effect."))
 
         # if we are going to force switching the backend, pull in
@@ -1358,11 +1355,11 @@ def use(arg, warn=False, force=True):
         if force:
             from matplotlib.pyplot import switch_backend
             switch_backend(name)
-    # Finally if pyplot is not imported update both rcParams and
-    # rcDefaults so restoring the defaults later with rcdefaults
-    # won't change the backend.  This is a bit of overkill as 'backend'
-    # is already in style.core.STYLE_BLACKLIST, but better to be safe.
     else:
+        # Finally if pyplot is not imported update both rcParams and
+        # rcDefaults so restoring the defaults later with rcdefaults
+        # won't change the backend.  This is a bit of overkill as 'backend'
+        # is already in style.core.STYLE_BLACKLIST, but better to be safe.
         rcParams['backend'] = rcParamsDefault['backend'] = name
 
 
