Skip to content

Commit 5407eee

Browse files
AniketsyAniketsy
committed
fix-lint
1 parent a3fb8e9 commit 5407eee

7 files changed

Lines changed: 85 additions & 25 deletions

File tree

‎src/numpy-stubs/_core/defchararray.pyi‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,6 @@ import numpy as np
77
from numpy import _OrderKACF as _Order # noqa: ICN003
88
from numpy._typing import _ArrayLikeAnyString_co as _ToAnyCharND, _ArrayLikeString_co as _ToStringND
99

10-
_ToShape: TypeAlias = _nt.ShapeLike
11-
1210
from ._multiarray_umath import compare_chararrays
1311
from .strings import (
1412
capitalize,
@@ -60,6 +58,8 @@ from .umath import (
6058
not_equal,
6159
)
6260

61+
_ToShape: TypeAlias = _nt.ShapeLike
62+
6363
__all__ = [
6464
"add",
6565
"array",

‎src/numpy-stubs/_core/numeric.pyi‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2031,7 +2031,9 @@ def count_nonzero(a: ArrayLike, axis: None = None, *, keepdims: L[False] = False
20312031
@overload
20322032
def count_nonzero(a: _nt.ToGeneric_0d, axis: _nt.ShapeLike | None = None, *, keepdims: L[True]) -> np.intp: ...
20332033
@overload
2034-
def count_nonzero(a: _nt.ToGeneric_1nd, axis: _nt.ShapeLike | None = None, *, keepdims: L[True]) -> _nt.Array[np.intp]: ...
2034+
def count_nonzero(
2035+
a: _nt.ToGeneric_1nd, axis: _nt.ShapeLike | None = None, *, keepdims: L[True]
2036+
) -> _nt.Array[np.intp]: ...
20352037
@overload
20362038
def count_nonzero(a: ArrayLike, axis: _nt.ShapeLike | None = None, *, keepdims: py_bool = False) -> Incomplete: ...
20372039

‎src/numpy-stubs/fft/_helper.pyi‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,9 @@ def fftshift(x: _nt.ToBool_1d, axes: _nt.ShapeLike | None = None) -> _nt.Array1D
2626
@overload
2727
def fftshift(x: _nt.ToBool_1nd, axes: _nt.ShapeLike | None = None) -> _nt.Array[np.bool]: ...
2828
@overload
29-
def fftshift(x: _nt.CanLenArray[_ScalarT, _ShapeT], axes: _nt.ShapeLike | None = None) -> _nt.Array[_ScalarT, _ShapeT]: ...
29+
def fftshift(
30+
x: _nt.CanLenArray[_ScalarT, _ShapeT], axes: _nt.ShapeLike | None = None
31+
) -> _nt.Array[_ScalarT, _ShapeT]: ...
3032
@overload
3133
def fftshift(x: _nt.ToInt_1d, axes: _nt.ShapeLike | None = None) -> _nt.Array1D[np.intp]: ...
3234
@overload

‎src/numpy-stubs/lib/_function_base_impl.pyi‎

Lines changed: 29 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -524,31 +524,49 @@ def gradient(
524524
) -> _nt.Array1D[_InexactTimeT] | Any: ...
525525
@overload # 1d, known inexact scalar-type
526526
def gradient(
527-
f: _nt.Array1D[_InexactTimeT], *varargs: _nt.CoComplex_nd, axis: _nt.ShapeLike | None = None, edge_order: L[1, 2] = 1
527+
f: _nt.Array1D[_InexactTimeT],
528+
*varargs: _nt.CoComplex_nd,
529+
axis: _nt.ShapeLike | None = None,
530+
edge_order: L[1, 2] = 1,
528531
) -> _nt.Array1D[_InexactTimeT]: ...
529532
@overload # 2d, known inexact scalar-type
530533
def gradient(
531-
f: _nt.Array2D[_InexactTimeT], *varargs: _nt.CoComplex_nd, axis: _nt.ShapeLike | None = None, edge_order: L[1, 2] = 1
534+
f: _nt.Array2D[_InexactTimeT],
535+
*varargs: _nt.CoComplex_nd,
536+
axis: _nt.ShapeLike | None = None,
537+
edge_order: L[1, 2] = 1,
532538
) -> _Mesh2[_InexactTimeT, _InexactTimeT]: ...
533539
@overload # 3d, known inexact scalar-type
534540
def gradient(
535-
f: _nt.Array3D[_InexactTimeT], *varargs: _nt.CoComplex_nd, axis: _nt.ShapeLike | None = None, edge_order: L[1, 2] = 1
541+
f: _nt.Array3D[_InexactTimeT],
542+
*varargs: _nt.CoComplex_nd,
543+
axis: _nt.ShapeLike | None = None,
544+
edge_order: L[1, 2] = 1,
536545
) -> _Mesh3[_InexactTimeT, _InexactTimeT, _InexactTimeT]: ...
537546
@overload # ?d, datetime64 scalar-type
538547
def gradient(
539548
f: _ArrayNoD[np.datetime64], *varargs: _nt.CoComplex_nd, axis: _nt.ShapeLike | None = None, edge_order: L[1, 2] = 1
540549
) -> _nt.Array1D[np.timedelta64] | tuple[_nt.Array[np.timedelta64], ...]: ...
541550
@overload # 1d, datetime64 scalar-type
542551
def gradient(
543-
f: _nt.Array1D[np.datetime64], *varargs: _nt.CoComplex_nd, axis: _nt.ShapeLike | None = None, edge_order: L[1, 2] = 1
552+
f: _nt.Array1D[np.datetime64],
553+
*varargs: _nt.CoComplex_nd,
554+
axis: _nt.ShapeLike | None = None,
555+
edge_order: L[1, 2] = 1,
544556
) -> _nt.Array1D[np.timedelta64]: ...
545557
@overload # 2d, datetime64 scalar-type
546558
def gradient(
547-
f: _nt.Array2D[np.datetime64], *varargs: _nt.CoComplex_nd, axis: _nt.ShapeLike | None = None, edge_order: L[1, 2] = 1
559+
f: _nt.Array2D[np.datetime64],
560+
*varargs: _nt.CoComplex_nd,
561+
axis: _nt.ShapeLike | None = None,
562+
edge_order: L[1, 2] = 1,
548563
) -> _Mesh2[np.timedelta64, np.timedelta64]: ...
549564
@overload # 3d, datetime64 scalar-type
550565
def gradient(
551-
f: _nt.Array3D[np.datetime64], *varargs: _nt.CoComplex_nd, axis: _nt.ShapeLike | None = None, edge_order: L[1, 2] = 1
566+
f: _nt.Array3D[np.datetime64],
567+
*varargs: _nt.CoComplex_nd,
568+
axis: _nt.ShapeLike | None = None,
569+
edge_order: L[1, 2] = 1,
552570
) -> _Mesh3[np.timedelta64, np.timedelta64, np.timedelta64]: ...
553571
@overload # 1d float-like
554572
def gradient(
@@ -1319,7 +1337,11 @@ def median(
13191337
) -> _nt.Array[np.float64]: ...
13201338
@overload # float array-like, axis=<given>
13211339
def median(
1322-
a: _nt.SequenceND[float], axis: _nt.ShapeLike, out: None = None, overwrite_input: bool = False, keepdims: bool = False
1340+
a: _nt.SequenceND[float],
1341+
axis: _nt.ShapeLike,
1342+
out: None = None,
1343+
overwrite_input: bool = False,
1344+
keepdims: bool = False,
13231345
) -> _nt.Array[np.float64]: ...
13241346
@overload # complex array-like, keepdims=True
13251347
def median(

‎src/numpy-stubs/ma/core.pyi‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1309,7 +1309,9 @@ class MaskedArray(np.ndarray[_ShapeT_co, _DTypeT_co]):
13091309
self, axis: _nt.ShapeLike | None = None, out: None = None, keepdims: bool | _NoValueType = ...
13101310
) -> bool_ | _nt.MArray[bool_]: ...
13111311
@overload
1312-
def all(self, axis: _nt.ShapeLike | None = None, *, out: _ArrayT, keepdims: bool | _NoValueType = ...) -> _ArrayT: ...
1312+
def all(
1313+
self, axis: _nt.ShapeLike | None = None, *, out: _ArrayT, keepdims: bool | _NoValueType = ...
1314+
) -> _ArrayT: ...
13131315
@overload
13141316
def all(self, axis: _nt.ShapeLike | None, out: _ArrayT, keepdims: bool | _NoValueType = ...) -> _ArrayT: ... # pyright: ignore[reportIncompatibleMethodOverride]
13151317

@@ -1326,7 +1328,9 @@ class MaskedArray(np.ndarray[_ShapeT_co, _DTypeT_co]):
13261328
self, axis: _nt.ShapeLike | None = None, out: None = None, keepdims: bool | _NoValueType = ...
13271329
) -> bool_ | _nt.MArray[bool_]: ...
13281330
@overload
1329-
def any(self, axis: _nt.ShapeLike | None = None, *, out: _ArrayT, keepdims: bool | _NoValueType = ...) -> _ArrayT: ...
1331+
def any(
1332+
self, axis: _nt.ShapeLike | None = None, *, out: _ArrayT, keepdims: bool | _NoValueType = ...
1333+
) -> _ArrayT: ...
13301334
@overload
13311335
def any(self, axis: _nt.ShapeLike | None, out: _ArrayT, keepdims: bool | _NoValueType = ...) -> _ArrayT: ... # pyright: ignore[reportIncompatibleMethodOverride]
13321336

‎src/numpy-stubs/random/_generator.pyi‎

Lines changed: 27 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -142,15 +142,25 @@ class Generator:
142142
def random(self, /, size: None = None, dtype: _DTypeLikeFloat = ..., out: None = None) -> float: ...
143143
@overload
144144
def random(
145-
self, /, size: _nt.ShapeLike | None = None, dtype: _nt.ToDTypeFloat64 = ..., *, out: _nt.Array[np.float64, _ShapeT]
145+
self,
146+
/,
147+
size: _nt.ShapeLike | None = None,
148+
dtype: _nt.ToDTypeFloat64 = ...,
149+
*,
150+
out: _nt.Array[np.float64, _ShapeT],
146151
) -> _nt.Array[np.float64, _ShapeT]: ...
147152
@overload
148153
def random(
149154
self, /, size: _nt.ShapeLike, dtype: _nt.ToDTypeFloat64 = ..., out: _nt.Array[np.float64] | None = None
150155
) -> _nt.Array[np.float64]: ...
151156
@overload
152157
def random(
153-
self, /, size: _nt.ShapeLike | None = None, dtype: _nt.ToDTypeFloat32 = ..., *, out: _nt.Array[np.float32, _ShapeT]
158+
self,
159+
/,
160+
size: _nt.ShapeLike | None = None,
161+
dtype: _nt.ToDTypeFloat32 = ...,
162+
*,
163+
out: _nt.Array[np.float32, _ShapeT],
154164
) -> _nt.Array[np.float32, _ShapeT]: ...
155165
@overload
156166
def random(
@@ -424,7 +434,9 @@ class Generator:
424434
@overload # size: (int, ...)
425435
def chisquare(self, /, df: _nt.CoFloating_nd, size: _nt.ShapeLike) -> _nt.Array[np.float64]: ...
426436
@overload # fallback
427-
def chisquare(self, /, df: _nt.CoFloating_nd, size: _nt.ShapeLike | None = None) -> float | _nt.Array[np.float64]: ...
437+
def chisquare(
438+
self, /, df: _nt.CoFloating_nd, size: _nt.ShapeLike | None = None
439+
) -> float | _nt.Array[np.float64]: ...
428440

429441
#
430442
@overload # workaround for microsoft/pyright#10232
@@ -558,7 +570,9 @@ class Generator:
558570
@overload # size: (int, ...)
559571
def standard_t(self, /, df: _nt.CoFloating_nd, size: _nt.ShapeLike) -> _nt.Array[np.float64]: ...
560572
@overload # fallback
561-
def standard_t(self, /, df: _nt.CoFloating_nd, size: _nt.ShapeLike | None = None) -> float | _nt.Array[np.float64]: ...
573+
def standard_t(
574+
self, /, df: _nt.CoFloating_nd, size: _nt.ShapeLike | None = None
575+
) -> float | _nt.Array[np.float64]: ...
562576

563577
#
564578
@overload # workaround for microsoft/pyright#10232
@@ -671,7 +685,9 @@ class Generator:
671685
@overload # size: None (default)
672686
def f(self, /, dfnum: _nt.CoFloating_0d, dfden: _nt.CoFloating_0d, size: None = None) -> float: ...
673687
@overload # size: (int, ...)
674-
def f(self, /, dfnum: _nt.CoFloating_nd, dfden: _nt.CoFloating_nd, size: _nt.ShapeLike) -> _nt.Array[np.float64]: ...
688+
def f(
689+
self, /, dfnum: _nt.CoFloating_nd, dfden: _nt.CoFloating_nd, size: _nt.ShapeLike
690+
) -> _nt.Array[np.float64]: ...
675691
@overload # fallback
676692
def f(
677693
self, /, dfnum: _nt.CoFloating_nd, dfden: _nt.CoFloating_nd, size: _nt.ShapeLike | None = None
@@ -720,7 +736,9 @@ class Generator:
720736
@overload # size: None (default)
721737
def wald(self, /, mean: float, scale: float, size: None = None) -> float: ...
722738
@overload # size: (int, ...)
723-
def wald(self, /, mean: _nt.CoFloating_nd, scale: _nt.CoFloating_nd, size: _nt.ShapeLike) -> _nt.Array[np.float64]: ...
739+
def wald(
740+
self, /, mean: _nt.CoFloating_nd, scale: _nt.CoFloating_nd, size: _nt.ShapeLike
741+
) -> _nt.Array[np.float64]: ...
724742
@overload # fallback
725743
def wald(
726744
self, /, mean: _nt.CoFloating_nd, scale: _nt.CoFloating_nd, size: _nt.ShapeLike | None = None
@@ -1237,7 +1255,9 @@ class Generator:
12371255
@overload # size: (int, ...) (keyword)
12381256
def poisson(self, /, lam: _nt.CoFloating_nd = 1.0, *, size: _nt.ShapeLike) -> _nt.Array[np.int64]: ...
12391257
@overload # fallback
1240-
def poisson(self, /, lam: _nt.CoFloating_nd = 1.0, size: _nt.ShapeLike | None = None) -> int | _nt.Array[np.int64]: ...
1258+
def poisson(
1259+
self, /, lam: _nt.CoFloating_nd = 1.0, size: _nt.ShapeLike | None = None
1260+
) -> int | _nt.Array[np.int64]: ...
12411261

12421262
#
12431263
@overload # workaround for microsoft/pyright#10232

‎src/numpy-stubs/random/mtrand.pyi‎

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -306,7 +306,9 @@ class RandomState:
306306
@overload # size: (int, ...)
307307
def chisquare(self, /, df: _nt.CoFloating_nd, size: _nt.ShapeLike) -> _nt.Array[np.float64]: ...
308308
@overload # fallback
309-
def chisquare(self, /, df: _nt.CoFloating_nd, size: _nt.ShapeLike | None = None) -> float | _nt.Array[np.float64]: ...
309+
def chisquare(
310+
self, /, df: _nt.CoFloating_nd, size: _nt.ShapeLike | None = None
311+
) -> float | _nt.Array[np.float64]: ...
310312

311313
#
312314
@overload # size: None (default)
@@ -382,7 +384,9 @@ class RandomState:
382384
@overload # size: (int, ...)
383385
def standard_t(self, /, df: _nt.CoFloating_nd, size: _nt.ShapeLike) -> _nt.Array[np.float64]: ...
384386
@overload # fallback
385-
def standard_t(self, /, df: _nt.CoFloating_nd, size: _nt.ShapeLike | None = None) -> float | _nt.Array[np.float64]: ...
387+
def standard_t(
388+
self, /, df: _nt.CoFloating_nd, size: _nt.ShapeLike | None = None
389+
) -> float | _nt.Array[np.float64]: ...
386390

387391
#
388392
@overload # size: None (default)
@@ -414,7 +418,9 @@ class RandomState:
414418
@overload # size: None (default)
415419
def f(self, /, dfnum: float, dfden: float, size: None = None) -> float: ...
416420
@overload # size: (int, ...)
417-
def f(self, /, dfnum: _nt.CoFloating_nd, dfden: _nt.CoFloating_nd, size: _nt.ShapeLike) -> _nt.Array[np.float64]: ...
421+
def f(
422+
self, /, dfnum: _nt.CoFloating_nd, dfden: _nt.CoFloating_nd, size: _nt.ShapeLike
423+
) -> _nt.Array[np.float64]: ...
418424
@overload # fallback
419425
def f(
420426
self, /, dfnum: _nt.CoFloating_nd, dfden: _nt.CoFloating_nd, size: _nt.ShapeLike | None = None
@@ -441,7 +447,9 @@ class RandomState:
441447
@overload # size: None (default)
442448
def wald(self, /, mean: float, scale: float, size: None = None) -> float: ...
443449
@overload # size: (int, ...)
444-
def wald(self, /, mean: _nt.CoFloating_nd, scale: _nt.CoFloating_nd, size: _nt.ShapeLike) -> _nt.Array[np.float64]: ...
450+
def wald(
451+
self, /, mean: _nt.CoFloating_nd, scale: _nt.CoFloating_nd, size: _nt.ShapeLike
452+
) -> _nt.Array[np.float64]: ...
445453
@overload # fallback
446454
def wald(
447455
self, /, mean: _nt.CoFloating_nd, scale: _nt.CoFloating_nd, size: _nt.ShapeLike | None = None
@@ -808,7 +816,9 @@ class RandomState:
808816
@overload # size: (int, ...) (keyword)
809817
def poisson(self, /, lam: _nt.CoFloating_nd = 1.0, *, size: _nt.ShapeLike) -> _nt.Array[np.int_]: ...
810818
@overload # fallback
811-
def poisson(self, /, lam: _nt.CoFloating_nd = 1.0, size: _nt.ShapeLike | None = None) -> int | _nt.Array[np.int_]: ...
819+
def poisson(
820+
self, /, lam: _nt.CoFloating_nd = 1.0, size: _nt.ShapeLike | None = None
821+
) -> int | _nt.Array[np.int_]: ...
812822

813823
#
814824
@overload # size: None (default)

0 commit comments

Comments
 (0)