@@ -4,7 +4,7 @@ from typing import Any, Literal, TypeVar, overload
44import numpy as np
55import optype as op
66import optype .numpy as onp
7- from numpy ._typing import _ArrayLike , _NestedSequence
7+ from numpy ._typing import _ArrayLike
88from scipy ._typing import AnyShape
99
1010_SCT = TypeVar ("_SCT" , bound = np .inexact [Any ])
@@ -41,21 +41,21 @@ def rfftfreq(n: int, d: float = 1.0, *, xp: ModuleType, device: object | None =
4141# TODO(jorenham): Array API support (for `x`)
4242# https://github.com/jorenham/scipy-stubs/issues/140
4343@overload
44- def fftshift (x : _ArrayLike [ _SCT ], axes : AnyShape | None = None ) -> onp .ArrayND [_SCT ]: ...
44+ def fftshift (x : onp . ToIntND | onp . SequenceND [ float ], axes : AnyShape | None = None ) -> onp .ArrayND [np . float64 ]: ...
4545@overload
46- def fftshift (x : onp .ToIntND | _NestedSequence [ float ], axes : AnyShape | None = None ) -> onp .ArrayND [np .float64 ]: ...
46+ def fftshift (x : onp .SequenceND [ complex ], axes : AnyShape | None = None ) -> onp .ArrayND [np . complex128 | np .float64 ]: ...
4747@overload
48- def fftshift (x : _NestedSequence [ complex ], axes : AnyShape | None = None ) -> onp .ArrayND [np . complex128 | np . float64 ]: ...
48+ def fftshift (x : _ArrayLike [ _SCT ], axes : AnyShape | None = None ) -> onp .ArrayND [_SCT ]: ...
4949@overload
5050def fftshift (x : onp .ToComplexND , axes : AnyShape | None = None ) -> onp .ArrayND [np .inexact [Any ]]: ...
5151
5252# TODO(jorenham): Array API support (for `x`)
5353# https://github.com/jorenham/scipy-stubs/issues/140
5454@overload
55- def ifftshift (x : _ArrayLike [ _SCT ], axes : AnyShape | None = None ) -> onp .ArrayND [_SCT ]: ...
55+ def ifftshift (x : onp . ToIntND | onp . SequenceND [ float ], axes : AnyShape | None = None ) -> onp .ArrayND [np . float64 ]: ...
5656@overload
57- def ifftshift (x : onp .ToIntND | _NestedSequence [ float ], axes : AnyShape | None = None ) -> onp .ArrayND [np .float64 ]: ...
57+ def ifftshift (x : onp .SequenceND [ complex ], axes : AnyShape | None = None ) -> onp .ArrayND [np . complex128 | np .float64 ]: ...
5858@overload
59- def ifftshift (x : _NestedSequence [ complex ], axes : AnyShape | None = None ) -> onp .ArrayND [np . complex128 | np . float64 ]: ...
59+ def ifftshift (x : _ArrayLike [ _SCT ], axes : AnyShape | None = None ) -> onp .ArrayND [_SCT ]: ...
6060@overload
6161def ifftshift (x : onp .ToComplexND , axes : AnyShape | None = None ) -> onp .ArrayND [np .inexact [Any ]]: ...
0 commit comments