Remove casting to array in EventCollection to preserve unit handling

afvincent · afvincent · commit 14c646077c5a · 2017-03-19T23:48:45.000+01:00
diff --git a/lib/matplotlib/collections.py b/lib/matplotlib/collections.py
@@ -1310,10 +1310,9 @@ def __init__(self,
 
         segment = (lineoffset + linelength / 2.,
                    lineoffset - linelength / 2.)
-        positions = np.asarray(positions)
-        if len(positions) == 0:
+        if positions is None or len(positions) == 0:
             segments = []
-        elif positions.ndim > 1:
+        elif hasattr(positions, 'ndim') and positions.ndim > 1:
             raise ValueError('positions cannot have a dimensionality greater '
                              'than 1 (in the ndarray sense)')
         elif (orientation is None or orientation.lower() == 'none' or
