_ufunc_config.pyi 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. from collections.abc import Callable
  2. from typing import Any, Literal, TypedDict
  3. from numpy import _SupportsWrite
  4. _ErrKind = Literal["ignore", "warn", "raise", "call", "print", "log"]
  5. _ErrFunc = Callable[[str, int], Any]
  6. class _ErrDict(TypedDict):
  7. divide: _ErrKind
  8. over: _ErrKind
  9. under: _ErrKind
  10. invalid: _ErrKind
  11. class _ErrDictOptional(TypedDict, total=False):
  12. all: None | _ErrKind
  13. divide: None | _ErrKind
  14. over: None | _ErrKind
  15. under: None | _ErrKind
  16. invalid: None | _ErrKind
  17. def seterr(
  18. all: None | _ErrKind = ...,
  19. divide: None | _ErrKind = ...,
  20. over: None | _ErrKind = ...,
  21. under: None | _ErrKind = ...,
  22. invalid: None | _ErrKind = ...,
  23. ) -> _ErrDict: ...
  24. def geterr() -> _ErrDict: ...
  25. def setbufsize(size: int) -> int: ...
  26. def getbufsize() -> int: ...
  27. def seterrcall(
  28. func: None | _ErrFunc | _SupportsWrite[str]
  29. ) -> None | _ErrFunc | _SupportsWrite[str]: ...
  30. def geterrcall() -> None | _ErrFunc | _SupportsWrite[str]: ...
  31. # See `numpy/__init__.pyi` for the `errstate` class and `no_nep5_warnings`