Skip to content

Commit c6acb2f

Browse files
authored
🏷️ special: stub the remaining private modules (#726)
2 parents 2d043db + f475046 commit c6acb2f

File tree

7 files changed

+381
-0
lines changed

7 files changed

+381
-0
lines changed

scipy-stubs/special/_comb.pyi

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
def _comb_int(N: int, k: int) -> int: ...

scipy-stubs/special/_ellip_harm_2.pyi

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
from typing import Final, TypedDict, type_check_only
2+
from typing_extensions import CapsuleType, ReadOnly
3+
4+
from scipy._lib._ccallback import LowLevelCallable as LowLevelCallable
5+
6+
@type_check_only
7+
class _CApiDict(TypedDict):
8+
_F_integrand: ReadOnly[CapsuleType]
9+
_F_integrand1: ReadOnly[CapsuleType]
10+
_F_integrand2: ReadOnly[CapsuleType]
11+
_F_integrand3: ReadOnly[CapsuleType]
12+
_F_integrand4: ReadOnly[CapsuleType]
13+
_set_action: ReadOnly[CapsuleType]
14+
15+
###
16+
17+
__pyx_capi__: Final[_CApiDict] = ...
18+
19+
nan: Final[float] = ...
20+
21+
def _ellipsoid(h2: float, k2: float, n: int, p: int, s: float) -> float: ...
22+
def _ellipsoid_norm(h2: float, k2: float, n: int, p: int) -> float: ...

scipy-stubs/special/_gufuncs.pyi

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
from typing import Final, Literal
2+
3+
import numpy as np
4+
5+
_lqmn: Final[np.ufunc] = ...
6+
_lqn: Final[np.ufunc] = ...
7+
_rctj: Final[np.ufunc] = ...
8+
_rcty: Final[np.ufunc] = ...
9+
10+
assoc_legendre_p_all: Final[dict[tuple[bool, Literal[0, 1, 2]], np.ufunc]] = ...
11+
legendre_p_all: Final[tuple[np.ufunc, np.ufunc, np.ufunc]] = ...
12+
sph_harm_y_all: Final[tuple[np.ufunc, np.ufunc, np.ufunc]] = ...
13+
sph_legendre_p_all: Final[tuple[np.ufunc, np.ufunc, np.ufunc]] = ...
14+
15+
def _set_action(code: int, action: int, /) -> None: ...
Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
from typing import SupportsIndex
2+
3+
def _nonneg_int_or_fail(n: SupportsIndex, var_name: str, strict: bool = True) -> int: ...

scipy-stubs/special/_specfun.pyi

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
import enum
2+
from typing import TypeAlias
3+
4+
import numpy as np
5+
import optype.numpy as onp
6+
7+
###
8+
9+
_Int1D: TypeAlias = onp.Array1D[np.int32]
10+
_Double1D: TypeAlias = onp.Array1D[np.float64]
11+
_CDouble1D: TypeAlias = onp.Array1D[np.complex128]
12+
13+
###
14+
15+
class Status(enum.IntEnum):
16+
OK = 0
17+
NoMemory = 1
18+
Other = 2
19+
20+
def airyzo(nt: int, kf: int) -> tuple[_Double1D, _Double1D, _Double1D, _Double1D]: ...
21+
def bernob(n: int) -> _Double1D: ...
22+
def cerzo(nt: int) -> _CDouble1D: ...
23+
def cpbdn(n: int, z: complex) -> tuple[_CDouble1D, _CDouble1D]: ...
24+
def cyzo(nt: int, kf: int, kc: int) -> tuple[_CDouble1D, _CDouble1D]: ...
25+
def eulerb(n: int) -> _Double1D: ...
26+
def fcoef(kd: int, m: int, q: float, a: float) -> _Double1D: ...
27+
def fcszo(kf: int, nt: int) -> _CDouble1D: ...
28+
def jdzo(nt: int) -> tuple[_Int1D, _Int1D, _Int1D, _Double1D]: ...
29+
def jyzo(n: int, nt: int) -> tuple[_Double1D, _Double1D, _Double1D, _Double1D]: ...
30+
def klvnzo(nt: int, kd: int) -> _Double1D: ...
31+
def lamn(n: int, x: float) -> tuple[int, _Double1D, _Double1D]: ...
32+
def lamv(v: float, x: float) -> tuple[float, _Double1D, _Double1D]: ...
33+
def pbdv(v: float, x: float) -> tuple[_Double1D, _Double1D, float, float]: ...
34+
def pbvv(v: float, x: float) -> tuple[_Double1D, _Double1D, float, float]: ...
35+
def sdmn(m: int, n: int, c: float, cv: float, kd: int) -> _Double1D: ...
36+
def segv(m: int, n: int, c: float, kd: int) -> tuple[float, _Double1D]: ...
Lines changed: 154 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,154 @@
1+
from typing import Final, Literal
2+
3+
import numpy as np
4+
5+
_cospi: Final[np.ufunc] = ...
6+
_iv_ratio: Final[np.ufunc] = ...
7+
_iv_ratio_c: Final[np.ufunc] = ...
8+
_lambertw: Final[np.ufunc] = ...
9+
_log1mexp: Final[np.ufunc] = ...
10+
_log1pmx: Final[np.ufunc] = ...
11+
_riemann_zeta: Final[np.ufunc] = ...
12+
_scaled_exp1: Final[np.ufunc] = ...
13+
_sinpi: Final[np.ufunc] = ...
14+
_spherical_in: Final[np.ufunc] = ...
15+
_spherical_in_d: Final[np.ufunc] = ...
16+
_spherical_jn: Final[np.ufunc] = ...
17+
_spherical_jn_d: Final[np.ufunc] = ...
18+
_spherical_kn: Final[np.ufunc] = ...
19+
_spherical_kn_d: Final[np.ufunc] = ...
20+
_spherical_yn: Final[np.ufunc] = ...
21+
_spherical_yn_d: Final[np.ufunc] = ...
22+
_zeta: Final[np.ufunc] = ...
23+
airy: Final[np.ufunc] = ...
24+
airye: Final[np.ufunc] = ...
25+
assoc_legendre_p: Final[dict[tuple[bool, Literal[0, 1, 2]], np.ufunc]] = ...
26+
bei: Final[np.ufunc] = ...
27+
beip: Final[np.ufunc] = ...
28+
ber: Final[np.ufunc] = ...
29+
berp: Final[np.ufunc] = ...
30+
besselpoly: Final[np.ufunc] = ...
31+
beta: Final[np.ufunc] = ...
32+
betaln: Final[np.ufunc] = ...
33+
binom: Final[np.ufunc] = ...
34+
cbrt: Final[np.ufunc] = ...
35+
cosdg: Final[np.ufunc] = ...
36+
cosm1: Final[np.ufunc] = ...
37+
cotdg: Final[np.ufunc] = ...
38+
dawsn: Final[np.ufunc] = ...
39+
ellipe: Final[np.ufunc] = ...
40+
ellipeinc: Final[np.ufunc] = ...
41+
ellipj: Final[np.ufunc] = ...
42+
ellipk: Final[np.ufunc] = ...
43+
ellipkinc: Final[np.ufunc] = ...
44+
ellipkm1: Final[np.ufunc] = ...
45+
erf: Final[np.ufunc] = ...
46+
erfc: Final[np.ufunc] = ...
47+
erfcx: Final[np.ufunc] = ...
48+
erfi: Final[np.ufunc] = ...
49+
exp1: Final[np.ufunc] = ...
50+
exp10: Final[np.ufunc] = ...
51+
exp2: Final[np.ufunc] = ...
52+
expi: Final[np.ufunc] = ...
53+
expit: Final[np.ufunc] = ...
54+
expm1: Final[np.ufunc] = ...
55+
exprel: Final[np.ufunc] = ...
56+
fresnel: Final[np.ufunc] = ...
57+
gamma: Final[np.ufunc] = ...
58+
gammainc: Final[np.ufunc] = ...
59+
gammaincc: Final[np.ufunc] = ...
60+
gammainccinv: Final[np.ufunc] = ...
61+
gammaincinv: Final[np.ufunc] = ...
62+
gammaln: Final[np.ufunc] = ...
63+
gammasgn: Final[np.ufunc] = ...
64+
hankel1: Final[np.ufunc] = ...
65+
hankel1e: Final[np.ufunc] = ...
66+
hankel2: Final[np.ufunc] = ...
67+
hankel2e: Final[np.ufunc] = ...
68+
hyp2f1: Final[np.ufunc] = ...
69+
i0: Final[np.ufunc] = ...
70+
i0e: Final[np.ufunc] = ...
71+
i1: Final[np.ufunc] = ...
72+
i1e: Final[np.ufunc] = ...
73+
it2i0k0: Final[np.ufunc] = ...
74+
it2j0y0: Final[np.ufunc] = ...
75+
it2struve0: Final[np.ufunc] = ...
76+
itairy: Final[np.ufunc] = ...
77+
iti0k0: Final[np.ufunc] = ...
78+
itj0y0: Final[np.ufunc] = ...
79+
itmodstruve0: Final[np.ufunc] = ...
80+
itstruve0: Final[np.ufunc] = ...
81+
iv: Final[np.ufunc] = ...
82+
ive: Final[np.ufunc] = ...
83+
j0: Final[np.ufunc] = ...
84+
j1: Final[np.ufunc] = ...
85+
jv: Final[np.ufunc] = ...
86+
jve: Final[np.ufunc] = ...
87+
k0: Final[np.ufunc] = ...
88+
k0e: Final[np.ufunc] = ...
89+
k1: Final[np.ufunc] = ...
90+
k1e: Final[np.ufunc] = ...
91+
kei: Final[np.ufunc] = ...
92+
keip: Final[np.ufunc] = ...
93+
kelvin: Final[np.ufunc] = ...
94+
ker: Final[np.ufunc] = ...
95+
kerp: Final[np.ufunc] = ...
96+
kv: Final[np.ufunc] = ...
97+
kve: Final[np.ufunc] = ...
98+
legendre_p: Final[tuple[np.ufunc, np.ufunc, np.ufunc]] = ...
99+
log1p: Final[np.ufunc] = ...
100+
log_expit: Final[np.ufunc] = ...
101+
log_ndtr: Final[np.ufunc] = ...
102+
log_wright_bessel: Final[np.ufunc] = ...
103+
loggamma: Final[np.ufunc] = ...
104+
logit: Final[np.ufunc] = ...
105+
mathieu_a: Final[np.ufunc] = ...
106+
mathieu_b: Final[np.ufunc] = ...
107+
mathieu_cem: Final[np.ufunc] = ...
108+
mathieu_modcem1: Final[np.ufunc] = ...
109+
mathieu_modcem2: Final[np.ufunc] = ...
110+
mathieu_modsem1: Final[np.ufunc] = ...
111+
mathieu_modsem2: Final[np.ufunc] = ...
112+
mathieu_sem: Final[np.ufunc] = ...
113+
modfresnelm: Final[np.ufunc] = ...
114+
modfresnelp: Final[np.ufunc] = ...
115+
modstruve: Final[np.ufunc] = ...
116+
ndtr: Final[np.ufunc] = ...
117+
obl_ang1: Final[np.ufunc] = ...
118+
obl_ang1_cv: Final[np.ufunc] = ...
119+
obl_cv: Final[np.ufunc] = ...
120+
obl_rad1: Final[np.ufunc] = ...
121+
obl_rad1_cv: Final[np.ufunc] = ...
122+
obl_rad2: Final[np.ufunc] = ...
123+
obl_rad2_cv: Final[np.ufunc] = ...
124+
pbdv: Final[np.ufunc] = ...
125+
pbvv: Final[np.ufunc] = ...
126+
pbwa: Final[np.ufunc] = ...
127+
pro_ang1: Final[np.ufunc] = ...
128+
pro_ang1_cv: Final[np.ufunc] = ...
129+
pro_cv: Final[np.ufunc] = ...
130+
pro_rad1: Final[np.ufunc] = ...
131+
pro_rad1_cv: Final[np.ufunc] = ...
132+
pro_rad2: Final[np.ufunc] = ...
133+
pro_rad2_cv: Final[np.ufunc] = ...
134+
psi: Final[np.ufunc] = ...
135+
radian: Final[np.ufunc] = ...
136+
rgamma: Final[np.ufunc] = ...
137+
sindg: Final[np.ufunc] = ...
138+
sph_harm: Final[np.ufunc] = ...
139+
sph_harm_y: Final[tuple[np.ufunc, np.ufunc, np.ufunc]] = ...
140+
sph_legendre_p: Final[tuple[np.ufunc, np.ufunc, np.ufunc]] = ...
141+
struve: Final[np.ufunc] = ...
142+
tandg: Final[np.ufunc] = ...
143+
voigt_profile: Final[np.ufunc] = ...
144+
wofz: Final[np.ufunc] = ...
145+
wright_bessel: Final[np.ufunc] = ...
146+
xlog1py: Final[np.ufunc] = ...
147+
xlogy: Final[np.ufunc] = ...
148+
y0: Final[np.ufunc] = ...
149+
y1: Final[np.ufunc] = ...
150+
yv: Final[np.ufunc] = ...
151+
yve: Final[np.ufunc] = ...
152+
zetac: Final[np.ufunc] = ...
153+
154+
def _set_action(code: int, action: int, /) -> None: ...

scipy-stubs/special/_ufuncs_cxx.pyi

Lines changed: 150 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,150 @@
1+
from typing import Final, TypedDict, type_check_only
2+
from typing_extensions import CapsuleType, ReadOnly
3+
4+
@type_check_only
5+
class _CApiDict(TypedDict):
6+
_export_beta_pdf_float: ReadOnly[CapsuleType]
7+
_export_beta_pdf_double: ReadOnly[CapsuleType]
8+
_export_beta_ppf_float: ReadOnly[CapsuleType]
9+
_export_beta_ppf_double: ReadOnly[CapsuleType]
10+
_export_binom_cdf_float: ReadOnly[CapsuleType]
11+
_export_binom_cdf_double: ReadOnly[CapsuleType]
12+
_export_binom_isf_float: ReadOnly[CapsuleType]
13+
_export_binom_isf_double: ReadOnly[CapsuleType]
14+
_export_binom_pmf_float: ReadOnly[CapsuleType]
15+
_export_binom_pmf_double: ReadOnly[CapsuleType]
16+
_export_binom_ppf_float: ReadOnly[CapsuleType]
17+
_export_binom_ppf_double: ReadOnly[CapsuleType]
18+
_export_binom_sf_float: ReadOnly[CapsuleType]
19+
_export_binom_sf_double: ReadOnly[CapsuleType]
20+
_export_cauchy_isf_float: ReadOnly[CapsuleType]
21+
_export_cauchy_isf_double: ReadOnly[CapsuleType]
22+
_export_cauchy_ppf_float: ReadOnly[CapsuleType]
23+
_export_cauchy_ppf_double: ReadOnly[CapsuleType]
24+
_export_hypergeom_cdf_float: ReadOnly[CapsuleType]
25+
_export_hypergeom_cdf_double: ReadOnly[CapsuleType]
26+
_export_hypergeom_mean_float: ReadOnly[CapsuleType]
27+
_export_hypergeom_mean_double: ReadOnly[CapsuleType]
28+
_export_hypergeom_pmf_float: ReadOnly[CapsuleType]
29+
_export_hypergeom_pmf_double: ReadOnly[CapsuleType]
30+
_export_hypergeom_sf_float: ReadOnly[CapsuleType]
31+
_export_hypergeom_sf_double: ReadOnly[CapsuleType]
32+
_export_hypergeom_skewness_float: ReadOnly[CapsuleType]
33+
_export_hypergeom_skewness_double: ReadOnly[CapsuleType]
34+
_export_hypergeom_variance_float: ReadOnly[CapsuleType]
35+
_export_hypergeom_variance_double: ReadOnly[CapsuleType]
36+
_export_invgauss_isf_float: ReadOnly[CapsuleType]
37+
_export_invgauss_isf_double: ReadOnly[CapsuleType]
38+
_export_invgauss_ppf_float: ReadOnly[CapsuleType]
39+
_export_invgauss_ppf_double: ReadOnly[CapsuleType]
40+
_export_landau_cdf_float: ReadOnly[CapsuleType]
41+
_export_landau_cdf_double: ReadOnly[CapsuleType]
42+
_export_landau_isf_float: ReadOnly[CapsuleType]
43+
_export_landau_isf_double: ReadOnly[CapsuleType]
44+
_export_landau_pdf_float: ReadOnly[CapsuleType]
45+
_export_landau_pdf_double: ReadOnly[CapsuleType]
46+
_export_landau_ppf_float: ReadOnly[CapsuleType]
47+
_export_landau_ppf_double: ReadOnly[CapsuleType]
48+
_export_landau_sf_float: ReadOnly[CapsuleType]
49+
_export_landau_sf_double: ReadOnly[CapsuleType]
50+
_export_nbinom_cdf_float: ReadOnly[CapsuleType]
51+
_export_nbinom_cdf_double: ReadOnly[CapsuleType]
52+
_export_nbinom_isf_float: ReadOnly[CapsuleType]
53+
_export_nbinom_isf_double: ReadOnly[CapsuleType]
54+
_export_nbinom_kurtosis_excess_float: ReadOnly[CapsuleType]
55+
_export_nbinom_kurtosis_excess_double: ReadOnly[CapsuleType]
56+
_export_nbinom_mean_float: ReadOnly[CapsuleType]
57+
_export_nbinom_mean_double: ReadOnly[CapsuleType]
58+
_export_nbinom_pmf_float: ReadOnly[CapsuleType]
59+
_export_nbinom_pmf_double: ReadOnly[CapsuleType]
60+
_export_nbinom_ppf_float: ReadOnly[CapsuleType]
61+
_export_nbinom_ppf_double: ReadOnly[CapsuleType]
62+
_export_nbinom_sf_float: ReadOnly[CapsuleType]
63+
_export_nbinom_sf_double: ReadOnly[CapsuleType]
64+
_export_nbinom_skewness_float: ReadOnly[CapsuleType]
65+
_export_nbinom_skewness_double: ReadOnly[CapsuleType]
66+
_export_nbinom_variance_float: ReadOnly[CapsuleType]
67+
_export_nbinom_variance_double: ReadOnly[CapsuleType]
68+
_export_ncf_isf_float: ReadOnly[CapsuleType]
69+
_export_ncf_isf_double: ReadOnly[CapsuleType]
70+
_export_ncf_kurtosis_excess_float: ReadOnly[CapsuleType]
71+
_export_ncf_kurtosis_excess_double: ReadOnly[CapsuleType]
72+
_export_ncf_mean_float: ReadOnly[CapsuleType]
73+
_export_ncf_mean_double: ReadOnly[CapsuleType]
74+
_export_ncf_pdf_float: ReadOnly[CapsuleType]
75+
_export_ncf_pdf_double: ReadOnly[CapsuleType]
76+
_export_ncf_sf_float: ReadOnly[CapsuleType]
77+
_export_ncf_sf_double: ReadOnly[CapsuleType]
78+
_export_ncf_skewness_float: ReadOnly[CapsuleType]
79+
_export_ncf_skewness_double: ReadOnly[CapsuleType]
80+
_export_ncf_variance_float: ReadOnly[CapsuleType]
81+
_export_ncf_variance_double: ReadOnly[CapsuleType]
82+
_export_nct_isf_float: ReadOnly[CapsuleType]
83+
_export_nct_isf_double: ReadOnly[CapsuleType]
84+
_export_nct_kurtosis_excess_float: ReadOnly[CapsuleType]
85+
_export_nct_kurtosis_excess_double: ReadOnly[CapsuleType]
86+
_export_nct_mean_float: ReadOnly[CapsuleType]
87+
_export_nct_mean_double: ReadOnly[CapsuleType]
88+
_export_nct_pdf_float: ReadOnly[CapsuleType]
89+
_export_nct_pdf_double: ReadOnly[CapsuleType]
90+
_export_nct_sf_float: ReadOnly[CapsuleType]
91+
_export_nct_sf_double: ReadOnly[CapsuleType]
92+
_export_nct_skewness_float: ReadOnly[CapsuleType]
93+
_export_nct_skewness_double: ReadOnly[CapsuleType]
94+
_export_nct_variance_float: ReadOnly[CapsuleType]
95+
_export_nct_variance_double: ReadOnly[CapsuleType]
96+
_export_ncx2_cdf_float: ReadOnly[CapsuleType]
97+
_export_ncx2_cdf_double: ReadOnly[CapsuleType]
98+
_export_ncx2_isf_float: ReadOnly[CapsuleType]
99+
_export_ncx2_isf_double: ReadOnly[CapsuleType]
100+
_export_ncx2_pdf_float: ReadOnly[CapsuleType]
101+
_export_ncx2_pdf_double: ReadOnly[CapsuleType]
102+
_export_ncx2_ppf_float: ReadOnly[CapsuleType]
103+
_export_ncx2_ppf_double: ReadOnly[CapsuleType]
104+
_export_ncx2_sf_float: ReadOnly[CapsuleType]
105+
_export_ncx2_sf_double: ReadOnly[CapsuleType]
106+
_export_skewnorm_cdf_float: ReadOnly[CapsuleType]
107+
_export_skewnorm_cdf_double: ReadOnly[CapsuleType]
108+
_export_skewnorm_isf_float: ReadOnly[CapsuleType]
109+
_export_skewnorm_isf_double: ReadOnly[CapsuleType]
110+
_export_skewnorm_ppf_float: ReadOnly[CapsuleType]
111+
_export_skewnorm_ppf_double: ReadOnly[CapsuleType]
112+
_export__stirling2_inexact: ReadOnly[CapsuleType]
113+
_export_ibeta_float: ReadOnly[CapsuleType]
114+
_export_ibeta_double: ReadOnly[CapsuleType]
115+
_export_ibetac_float: ReadOnly[CapsuleType]
116+
_export_ibetac_double: ReadOnly[CapsuleType]
117+
_export_ibetac_inv_float: ReadOnly[CapsuleType]
118+
_export_ibetac_inv_double: ReadOnly[CapsuleType]
119+
_export_ibeta_inv_float: ReadOnly[CapsuleType]
120+
_export_ibeta_inv_double: ReadOnly[CapsuleType]
121+
_export_fellint_RC: ReadOnly[CapsuleType]
122+
_export_cellint_RC: ReadOnly[CapsuleType]
123+
_export_fellint_RD: ReadOnly[CapsuleType]
124+
_export_cellint_RD: ReadOnly[CapsuleType]
125+
_export_fellint_RF: ReadOnly[CapsuleType]
126+
_export_cellint_RF: ReadOnly[CapsuleType]
127+
_export_fellint_RG: ReadOnly[CapsuleType]
128+
_export_cellint_RG: ReadOnly[CapsuleType]
129+
_export_fellint_RJ: ReadOnly[CapsuleType]
130+
_export_cellint_RJ: ReadOnly[CapsuleType]
131+
_export_erfinv_float: ReadOnly[CapsuleType]
132+
_export_erfinv_double: ReadOnly[CapsuleType]
133+
_export_hyp1f1_double: ReadOnly[CapsuleType]
134+
_export_ncf_cdf_float: ReadOnly[CapsuleType]
135+
_export_ncf_cdf_double: ReadOnly[CapsuleType]
136+
_export_ncf_ppf_float: ReadOnly[CapsuleType]
137+
_export_ncf_ppf_double: ReadOnly[CapsuleType]
138+
_export_nct_cdf_float: ReadOnly[CapsuleType]
139+
_export_nct_cdf_double: ReadOnly[CapsuleType]
140+
_export_nct_ppf_float: ReadOnly[CapsuleType]
141+
_export_nct_ppf_double: ReadOnly[CapsuleType]
142+
_export_powm1_float: ReadOnly[CapsuleType]
143+
_export_powm1_double: ReadOnly[CapsuleType]
144+
_export_wrightomega: ReadOnly[CapsuleType]
145+
_export_wrightomega_real: ReadOnly[CapsuleType]
146+
_set_action: ReadOnly[CapsuleType]
147+
148+
###
149+
150+
__pyx_capi__: Final[_CApiDict] = ...

0 commit comments

Comments
 (0)