Move seldom used _is_array func to lambda

lukelbd · lukelbd · commit a2cf7b42e943 · 2022-05-29T21:08:36.000-06:00
diff --git a/proplot/axes/plot.py b/proplot/axes/plot.py
@@ -4181,10 +4181,8 @@ def _iter_arg_cols(self, *args, label=None, labels=None, values=None, **kwargs):
         # Handle cycle args and label lists
         # NOTE: Arrays here should have had metadata stripped by _parse_1d_args
         # but could still be pint quantities that get processed by axis converter.
-        n = max(
-            1 if not inputs._is_array(a) or a.ndim < 2 else a.shape[-1]
-            for a in args
-        )
+        is_array = lambda data: hasattr(data, 'ndim') and hasattr(data, 'shape')  # noqa: E731, E501
+        n = max(1 if not is_array(a) or a.ndim < 2 else a.shape[-1] for a in args)
         labels = _not_none(label=label, values=values, labels=labels)
         if not np.iterable(labels) or isinstance(labels, str):
             labels = n * [labels]
@@ -4202,9 +4200,7 @@ def _iter_arg_cols(self, *args, label=None, labels=None, values=None, **kwargs):
         for i in range(n):
             kw = kwargs.copy()
             kw['label'] = labels[i] or None
-            a = tuple(
-                a if not inputs._is_array(a) or a.ndim < 2 else a[..., i] for a in args
-            )
+            a = tuple(a if not is_array(a) or a.ndim < 2 else a[..., i] for a in args)
             yield (i, n, *a, kw)
 
     # Related parsing functions for warnings
diff --git a/proplot/internals/inputs.py b/proplot/internals/inputs.py
@@ -54,16 +54,6 @@ def _load_objects():
 
 
 # Type utilities
-def _is_array(data):
-    """
-    Test whether input is numpy array or pint quantity.
-    """
-    # NOTE: This is used in _iter_columns to identify 2D matrices that
-    # should be iterated over and omit e.g. scalar marker size or marker color.
-    _load_objects()
-    return isinstance(data, ndarray) or ndarray is not Quantity and isinstance(data, Quantity)  # noqa: E501
-
-
 def _is_numeric(data):
     """
     Test whether input is numeric array rather than datetime or strings.
