docarray · samsja · Mar 1, 2023 · Feb 7, 2023 · Feb 8, 2023 · Jan 23, 2023
diff --git a/docarray/array/array/array.py b/docarray/array/array/array.py
@@ -23,7 +23,7 @@
 from docarray.array.array.io import IOMixinArray
 from docarray.base_document import AnyDocument, BaseDocument
 from docarray.typing import NdArray
-from docarray.utils.misc import is_torch_available
+from docarray.utils.misc import is_np_int, is_torch_available
 
 if TYPE_CHECKING:
     from pydantic import BaseConfig
@@ -56,17 +56,6 @@ def _delegate_meth(self, *args, **kwargs):
     return _delegate_meth
 
 
-def _is_np_int(item: Any) -> bool:
-    dtype = getattr(item, 'dtype', None)
-    ndim = getattr(item, 'ndim', None)
-    if dtype is not None and ndim is not None:
-        try:
-            return ndim == 0 and np.issubdtype(dtype, np.integer)
-        except TypeError:
-            return False
-    return False  # this is unreachable, but mypy wants it
-
-
 class DocumentArray(IOMixinArray, AnyDocumentArray[T_doc]):
     """
      DocumentArray is a container of Documents.
@@ -255,7 +244,7 @@ def _normalize_index_item(
             return item
 
         # numpy index types
-        if _is_np_int(item):
+        if is_np_int(item):
             return item.item()
 
         index_has_getitem = hasattr(item, '__getitem__')

diff --git a/docarray/doc_index/__init__.py b/docarray/doc_index/__init__.py