Merge pull request #220 from tino-michael/astropy_units

drawing no longer breaks if data contain astropy quantities

Author: nschloe

matplotlib2tikz/line2d.py

@@ -128,6 +128,17 @@ def draw_line2d(data, obj):
     # xdata, ydata = _transform_to_data_coordinates(obj, *obj.get_data())
     xdata, ydata = obj.get_data()
 
+    # matplotlib allows plotting of data containing `astropy.units`, but they will
+    # break the formatted string here. Try to strip the units from the data.
+    try:
+        xdata = xdata.value
+    except AttributeError:
+        pass
+    try:
+        ydata = ydata.value
+    except AttributeError:
+        pass
+
     try:
         has_mask = ydata.mask.any()
     except AttributeError: