JRE - Added unit code to PolarAxes rgrids and thetagrids

James R. Evans · James R. Evans · commit 2bd3ed762acf · 2012-01-31T10:14:55.000-08:00
diff --git a/lib/matplotlib/projections/polar.py b/lib/matplotlib/projections/polar.py
@@ -481,6 +481,8 @@ def set_thetagrids(self, angles, labels=None, frac=None, fmt=None,
 
         ACCEPTS: sequence of floats
         """
+        # Make sure we take into account unitized data
+        angles = self.convert_yunits(angles)
         angles = np.asarray(angles, np.float_)
         self.set_xticks(angles * (np.pi / 180.0))
         if labels is not None:
@@ -521,6 +523,8 @@ def set_rgrids(self, radii, labels=None, angle=None, rpad=None, fmt=None,
 
         ACCEPTS: sequence of floats
         """
+        # Make sure we take into account unitized data
+        radii = self.convert_xunits(radii)
         radii = np.asarray(radii)
         rmin = radii.min()
         if rmin <= 0:
