Merge pull request #15925 from anntzer/units

timhoffm · web-flow · commit db7cd7e84ba8 · 2019-12-13T12:26:02.000+01:00
Optimize setting units to None when they're already None.
diff --git a/lib/matplotlib/axis.py b/lib/matplotlib/axis.py
@@ -1508,18 +1508,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):
