Skip to content

Commit

Permalink
signal: Change type annotation for axis from int to op.CanIndex
Browse files Browse the repository at this point in the history
In `_savitzy_golay.pyi`.
  • Loading branch information
pavyamsiri committed Oct 21, 2024
1 parent d25d475 commit c582f6f
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion scipy-stubs/signal/_savitzky_golay.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ from typing import Literal, TypeAlias

import numpy as np
import numpy.typing as npt
import optype as op

_Array_fc_1d: TypeAlias = np.ndarray[tuple[int], np.dtype[np.inexact[npt.NBitBase]]]
_Mode: TypeAlias = Literal["mirror", "constant", "nearest", "wrap", "interp"]
Expand All @@ -20,7 +21,7 @@ def savgol_filter(
polyorder: int,
deriv: int = 0,
delta: float = 1.0,
axis: int = -1,
axis: op.CanIndex = -1,
mode: _Mode = "interp",
cval: float = 0.0,
) -> npt.NDArray[np.float32 | np.float64]: ...

0 comments on commit c582f6f

Please sign in to comment.