mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Optional to boundary arg of stft.

Browse Source
This commit is contained in:
Jiayu Chang 2022-12-30 22:42:45 -05:00
parent f624b6e751
commit 8df97ff2a3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/_src/scipy/signal.py
View File

@ -439,7 +439,7 @@ def _spectral_helper(x: Array, y: Optional[ArrayLike], fs: ArrayLike = 1.0,
@_wraps(osp_signal.stft)
def stft(x: Array, fs: ArrayLike = 1.0, window: str = 'hann', nperseg: int = 256,
noverlap: Optional[int] = None, nfft: Optional[int] = None,
detrend: bool = False, return_onesided: bool = True, boundary: str = 'zeros',
detrend: bool = False, return_onesided: bool = True, boundary: Optional[str] = 'zeros',
padded: bool = True, axis: int = -1) -> Tuple[Array, Array, Array]:
return _spectral_helper(x, None, fs, window, nperseg, noverlap,
nfft, detrend, return_onesided,