Minor code cleanups in basic_units example

efiring · efiring · commit aa6f6ee9d954 · 2018-12-11T11:31:39.000-10:00
diff --git a/examples/units/basic_units.py b/examples/units/basic_units.py
@@ -348,9 +348,9 @@ def convert(val, unit, axis):
         if units.ConversionInterface.is_numlike(val):
             return val
         if np.iterable(val):
-            if np.ma.isMaskedArray(val):
+            if isinstance(val, np.ma.MaskedArray):
                 val = val.astype(float).filled(np.nan)
-            out = np.empty((len(val),),  dtype=float)
+            out = np.empty(len(val))
             for i, thisval in enumerate(val):
                 if np.ma.is_masked(thisval):
                     out[i] = np.nan
