Merge pull request #15930 from meeseeksmachine/auto-backport-of-pr-15925-on-v3.2.x

timhoffm · web-flow · commit fe93c1870b24 · 2019-12-13T13:57:41.000+01:00
Backport PR #15925 on branch v3.2.x (Optimize setting units to None when they're already None.)
diff --git a/lib/matplotlib/axis.py b/lib/matplotlib/axis.py
@@ -1584,18 +1584,12 @@ def set_units(self, u):
         ----------
         u : units tag
         """
-        pchanged = False
-        if u is None:
-            self.units = None
-            pchanged = True
-        else:
-            if u != self.units:
-                self.units = u
-                pchanged = True
-        if pchanged:
-            self._update_axisinfo()
-            self.callbacks.process('units')
-            self.callbacks.process('units finalize')
+        if u == self.units:
+            return
+        self.units = u
+        self._update_axisinfo()
+        self.callbacks.process('units')
+        self.callbacks.process('units finalize')
         self.stale = True
 
     def get_units(self):
