update stub files

- Added default values in stub files.

- Added ModeInt type union to allow integer values where MODE enum
was expected.

- Used upper bounds for DataT and CDataT types instead of variable
constraints.

Author: NiklasMueller2001

pywt/_extensions/_dwt.pyi

@@ -1,12 +1,12 @@
 from numpy.typing import NDArray
 
-from pywt import MODE, CDataT, Wavelet
+from ._pywt import CDataT, ModeInt, Wavelet
 
 def dwt_max_level(data_len: int, filter_len: int) -> int: ...
-def dwt_coeff_len(size_t: int, filter_len: int, mode: MODE) -> int: ...
-def dwt_single(data: NDArray[CDataT], wavelet: Wavelet, mode: MODE) -> tuple[NDArray[CDataT], NDArray[CDataT]]: ...
-def dwt_axis(data: NDArray[CDataT], wavelet: Wavelet, mode: MODE, axis: int = ...) -> tuple[NDArray[CDataT], NDArray[CDataT]]: ...
-def idwt_single(cA: NDArray[CDataT], cD: NDArray[CDataT], wavelet: Wavelet, mode: MODE) -> NDArray[CDataT]: ...
-def idwt_axis(coefs_a: NDArray[CDataT], coefs_d: NDArray[CDataT], wavelet: Wavelet, mode: MODE, axis: int = ...) -> NDArray[CDataT]: ...
+def dwt_coeff_len(size_t: int, filter_len: int, mode: ModeInt) -> int: ...
+def dwt_single(data: NDArray[CDataT], wavelet: Wavelet, mode: ModeInt) -> tuple[NDArray[CDataT], NDArray[CDataT]]: ...
+def dwt_axis(data: NDArray[CDataT], wavelet: Wavelet, mode: ModeInt, axis: int = 0) -> tuple[NDArray[CDataT], NDArray[CDataT]]: ...
+def idwt_single(cA: NDArray[CDataT], cD: NDArray[CDataT], wavelet: Wavelet, mode: ModeInt) -> NDArray[CDataT]: ...
+def idwt_axis(coefs_a: NDArray[CDataT], coefs_d: NDArray[CDataT], wavelet: Wavelet, mode: ModeInt, axis: int = 0) -> NDArray[CDataT]: ...
 def upcoef(do_rec_a: bool, coeffs: NDArray[CDataT], wavelet: Wavelet, level: int, take: int) -> NDArray[CDataT]: ...
-def downcoef(do_dec_a: bool, data: NDArray[CDataT], wavelet: Wavelet, mode: MODE, level: int) -> NDArray[CDataT]: ...
+def downcoef(do_dec_a: bool, data: NDArray[CDataT], wavelet: Wavelet, mode: ModeInt, level: int) -> NDArray[CDataT]: ...

pywt/_extensions/_pywt.pyi

@@ -1,5 +1,5 @@
 from enum import IntEnum
-from typing import Any, Literal, Optional, TypeVar
+from typing import Any, Literal, Optional, TypeAlias, TypeVar
 
 import numpy as np
 
@@ -20,17 +20,13 @@ _WaveletFamily = Literal[
     "cmor",
 ]
 
-DataT = TypeVar("DataT", np.float32, np.float64)
+DataT = TypeVar("DataT", bound=np.float32 | np.float64)
 
 CDataT = TypeVar(
-    "CDataT",
-    np.float32,
-    np.float64,
-    np.complex64,
-    np.complex128,
+    "CDataT", bound=np.float32 | np.float64 | np.complex64 | np.complex128
 )
 
-_Kind = Literal["all", "continuous", "discrete"]
+_Kind: TypeAlias = Literal["all", "continuous", "discrete"]
 
 _Symmetry = Literal[
     "asymmetric",
@@ -53,6 +49,8 @@ class MODE(IntEnum):
     MODE_ANTIREFLECT = 8
     MODE_MAX = 9
 
+ModeInt = MODE | Literal[-1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
+
 ModeName = Literal[
     "zero",
     "constant",
@@ -84,11 +82,11 @@ class _Modes:
 
 Modes = _Modes()
 
-def wavelist(family: _WaveletFamily | None = ..., kind: _Kind = ...) -> list[str]: ...
-def families(short: bool = ...) -> list[str]: ...
+def wavelist(family: _WaveletFamily | None = None, kind: _Kind = "all") -> list[str]: ...
+def families(short: bool = True) -> list[str]: ...
 
 class Wavelet:
-    def __init__(self, name: str = ..., filter_bank: Any = ...) -> None: ...
+    def __init__(self, name: str = "", filter_bank: Any = None) -> None: ...
     def __len__(self) -> int: ...
     @property
     def name(self) -> str: ...
@@ -140,7 +138,7 @@ class Wavelet:
     ) -> tuple[list[float], list[float], list[float], list[float]]: ...
 
 class ContinuousWavelet:
-    def __init__(self, name: str = ..., dtype: DataT = ...) -> None: ...
+    def __init__(self, name: str = "", dtype: DataT = np.float64) -> None: ...
     @property
     def family_number(self) -> int: ...
     @property
@@ -182,4 +180,6 @@ class ContinuousWavelet:
     @property
     def symmetry(self) -> _Symmetry: ...
 
-def DiscreteContinuousWavelet(name: str = ..., filter_bank: Any = ...) -> Wavelet | ContinuousWavelet : ...
+def DiscreteContinuousWavelet(
+    name: str = "", filter_bank: Any = None
+) -> Wavelet | ContinuousWavelet: ...

pywt/_extensions/_swt.pyi

@@ -1,7 +1,7 @@
 from numpy.typing import NDArray
 
-from pywt import CDataT, Wavelet
+from ._pywt import CDataT, Wavelet
 
 def swt_max_level(input_len: int) -> int: ...
-def swt(data: NDArray[CDataT], wavelet: Wavelet, level: int, start_level: int, trim_approx: bool = ...) -> NDArray[CDataT]: ...
-def swt_axis(data: NDArray[CDataT], wavelet: Wavelet, level: int, start_level: int, axis: int = ..., trim_approx: bool = ...) -> NDArray[CDataT]: ...
+def swt(data: NDArray[CDataT], wavelet: Wavelet, level: int, start_level: int, trim_approx: bool = False) -> NDArray[CDataT]: ...
+def swt_axis(data: NDArray[CDataT], wavelet: Wavelet, level: int, start_level: int, axis: int = 0, trim_approx: bool = False) -> NDArray[CDataT]: ...