from collections.abc import Iterable

from typing import Any, Final, overload

import numpy as np

from numpy import _CastingKind

###

class UFuncTypeError(TypeError):

ufunc: Final[np.ufunc]

def __init__(self, /, ufunc: np.ufunc) -> None: ...

class _UFuncNoLoopError(UFuncTypeError):

dtypes: tuple[np.dtype, ...]

def __init__(self, /, ufunc: np.ufunc, dtypes: Iterable[np.dtype]) -> None: ...

class _UFuncBinaryResolutionError(_UFuncNoLoopError):

dtypes: tuple[np.dtype, np.dtype] # pyrefly: ignore[bad-override]

def __init__(self, /, ufunc: np.ufunc, dtypes: Iterable[np.dtype]) -> None: ...

class _UFuncCastingError(UFuncTypeError):

casting: Final[_CastingKind]

from_: Final[np.dtype]

to: Final[np.dtype]

def __init__(self, /, ufunc: np.ufunc, casting: _CastingKind, from_: np.dtype, to: np.dtype) -> None: ...

class _UFuncInputCastingError(_UFuncCastingError):

in_i: Final[int]

def __init__(self, /, ufunc: np.ufunc, casting: _CastingKind, from_: np.dtype, to: np.dtype, i: int) -> None: ...

class _UFuncOutputCastingError(_UFuncCastingError):

out_i: Final[int]

def __init__(self, /, ufunc: np.ufunc, casting: _CastingKind, from_: np.dtype, to: np.dtype, i: int) -> None: ...

class _ArrayMemoryError(MemoryError):

shape: tuple[int, ...]

dtype: np.dtype

def __init__(self, /, shape: tuple[int, ...], dtype: np.dtype) -> None: ...

@property

def _total_size(self) -> int: ...

@staticmethod

def _size_to_string(num_bytes: int) -> str: ...

@overload

def _unpack_tuple[T](tup: tuple[T]) -> T: ...

@overload

def _unpack_tuple[TupleT: tuple[()] | tuple[Any, Any, *tuple[Any, ...]]](tup: TupleT) -> TupleT: ...

def _display_as_base[ExceptionT: Exception](cls: type[ExceptionT]) -> type[ExceptionT]: ...