Speed up Vector init from list

otsaloma · Dec 10, 2024 · abbeefe · abbeefe
1 parent a8a7a84
commit abbeefe
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.flake8 b/.flake8
@@ -1,4 +1,4 @@
 [flake8]
 select = E1,E9,F
-ignore = E129
+ignore = E125,E129
 exclude = doc/comparison/blocks,venv
diff --git a/dataiter/vector.py b/dataiter/vector.py
@@ -442,7 +442,9 @@ def _np_array(cls, object, dtype=None):
         # NumPy still defaults to fixed width strings.
         # In some cases we can only fix the dtype ex-post.
         if dtype is None:
-            if util.unique_types(object) == {str}:
+            if object and (
+                isinstance(object[0], str) and
+                isinstance(object[-1], str)):
                 dtype = dtypes.string
         dtype = cls._map_input_dtype(dtype)
         array = np.array(object, dtype)