arraypad.pyi 694 B

12345678910111213141516171819202122
  1. from collections.abc import Mapping
  2. from typing import Any, SupportsIndex
  3. import numpy as np
  4. import numpy.typing as npt
  5. def mode_func(
  6. ar: npt.NDArray[np.number[Any]],
  7. width: tuple[int, int],
  8. iaxis: SupportsIndex,
  9. kwargs: Mapping[str, Any],
  10. ) -> None: ...
  11. AR_i8: npt.NDArray[np.int64]
  12. AR_f8: npt.NDArray[np.float64]
  13. AR_LIKE: list[int]
  14. reveal_type(np.pad(AR_i8, (2, 3), "constant")) # E: ndarray[Any, dtype[{int64}]]
  15. reveal_type(np.pad(AR_LIKE, (2, 3), "constant")) # E: ndarray[Any, dtype[Any]]
  16. reveal_type(np.pad(AR_f8, (2, 3), mode_func)) # E: ndarray[Any, dtype[{float64}]]
  17. reveal_type(np.pad(AR_f8, (2, 3), mode_func, a=1, b=2)) # E: ndarray[Any, dtype[{float64}]]