Warning
This package is in early development and is not yet stable. It is intended as a typing toolkit for working with the Python array API standard, but it may change significantly in the future.
pip install array-api-typing
using uv
uv add array-api-typing
from source, using pip
pip install git+https://github.com/data-apis/array-api-typing.git
building from source
cd /path/to/parent git clone https://github.com/data-apis/array-api-typing.git cd array-api-typing pip install -e . # editable mode
>>> import array_api_typing as xpt >>> import numpy as np >>> def func(x: xpt.HasArrayNamespace) -> xpt.HasArrayNamespace: ... return x >>> func(np.array([1, 2, 3])) array([1, 2, 3])