__all__ = ("ArrayType",)

from typing import TypeVar

import numpy as np
from jaxtyping import Array

# NOTE: we declare this in another module to be able to patch it when building the docs
ArrayType = TypeVar(
    "ArrayType",
    Array,
    np.ndarray,
)
