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-typingusing uv
uv add array-api-typingfrom source, using pip
pip install git+https://github.com/data-apis/array-api-typing.gitbuilding 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])