_ufunc_config.pyi 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. import sys
  2. from typing import Optional, Union, Callable, Any
  3. if sys.version_info >= (3, 8):
  4. from typing import Literal, Protocol, TypedDict
  5. else:
  6. from typing_extensions import Literal, Protocol, TypedDict
  7. _ErrKind = Literal["ignore", "warn", "raise", "call", "print", "log"]
  8. _ErrFunc = Callable[[str, int], Any]
  9. class _SupportsWrite(Protocol):
  10. def write(self, __msg: str) -> Any: ...
  11. class _ErrDict(TypedDict):
  12. divide: _ErrKind
  13. over: _ErrKind
  14. under: _ErrKind
  15. invalid: _ErrKind
  16. class _ErrDictOptional(TypedDict, total=False):
  17. all: Optional[_ErrKind]
  18. divide: Optional[_ErrKind]
  19. over: Optional[_ErrKind]
  20. under: Optional[_ErrKind]
  21. invalid: Optional[_ErrKind]
  22. def seterr(
  23. all: Optional[_ErrKind] = ...,
  24. divide: Optional[_ErrKind] = ...,
  25. over: Optional[_ErrKind] = ...,
  26. under: Optional[_ErrKind] = ...,
  27. invalid: Optional[_ErrKind] = ...,
  28. ) -> _ErrDict: ...
  29. def geterr() -> _ErrDict: ...
  30. def setbufsize(size: int) -> int: ...
  31. def getbufsize() -> int: ...
  32. def seterrcall(
  33. func: Union[None, _ErrFunc, _SupportsWrite]
  34. ) -> Union[None, _ErrFunc, _SupportsWrite]: ...
  35. def geterrcall() -> Union[None, _ErrFunc, _SupportsWrite]: ...
  36. # See `numpy/__init__.pyi` for the `errstate` class