Skip to content
Snippets Groups Projects
Commit f5a5ca63 authored by Felix Fischer's avatar Felix Fischer :shrimp:
Browse files

Merge branch 'fix_type' into 'master'

Fix type

See merge request !56
parents 17cb1773 fbc2abe9
No related branches found
No related tags found
1 merge request!56Fix type
Pipeline #1377846 passed
...@@ -5,4 +5,4 @@ useful across several different projects and repositories. It also contains ...@@ -5,4 +5,4 @@ useful across several different projects and repositories. It also contains
stub-files with several data-type annotations for scipy and matplot-lib stub-files with several data-type annotations for scipy and matplot-lib
functions.""" functions."""
__version__ = "1.8.4" __version__ = "1.8.5"
...@@ -3,11 +3,14 @@ ...@@ -3,11 +3,14 @@
Stub-files for scipy.signal. This module contains the type-annotations for Stub-files for scipy.signal. This module contains the type-annotations for
scipy's library for fourier-transformations and signal analysis. scipy's library for fourier-transformations and signal analysis.
""" """
from typing import Tuple, Optional, Union, overload, TypeVar from typing import Tuple, Optional, Union, overload, TypeVar, Any
from scientific_plots.types_ import Vector, Matrix, Tensor from scientific_plots.types_ import Vector, Matrix, Tensor
Input = TypeVar("Input", Vector, Matrix, Tensor)
def welch( def welch(
Y: Union[Vector, list[float]], Y: Union[Vector, list[float]],
fs: Optional[float] = None, fs: Optional[float] = None,
...@@ -47,17 +50,17 @@ def savgol_filter( ...@@ -47,17 +50,17 @@ def savgol_filter(
def sosfilt( def sosfilt(
sos: Single, x: Union[Single, Matrix], sos: Single, x: Input,
axis: int = -1, zi: Optional[Single] = None, axis: int = -1, zi: Optional[Single] = None,
padtype: Optional[str] = "odd", padlen: Optional[int] = None)\ padtype: Optional[str] = "odd", padlen: Optional[int] = None)\
-> Vector: ... -> Input: ...
def sosfiltfilt( def sosfiltfilt(
sos: Single, x: Union[Single, Matrix], sos: Single, x: Input,
axis: int = -1, zi: Optional[Single] = None, axis: int = -1, zi: Optional[Single] = None,
padtype: Optional[str] = "odd", padlen: Optional[int] = None)\ padtype: Optional[str] = "odd", padlen: Optional[int] = None)\
-> Vector: ... -> Input: ...
def butter( def butter(
...@@ -65,10 +68,14 @@ def butter( ...@@ -65,10 +68,14 @@ def butter(
output: str = "ba", fs: Optional[float] = None) -> Vector: ... output: str = "ba", fs: Optional[float] = None) -> Vector: ...
Input = TypeVar("Input", Vector, Matrix, Tensor)
def wiener( def wiener(
im: Input, im: Input,
noise: Optional[float] = None, noise: Optional[float] = None,
mysize: Optional[Union[int, Vector]] = None) -> Input: ... mysize: Optional[Union[int, Vector]] = None) -> Input: ...
def sosfreqz(sos: Any,
worN: Optional[Union[int, Vector]] = 512,
whole: bool = False,
fs: Optional[float] = None)\
-> tuple[Vector, Vector]: ...
...@@ -13,10 +13,3 @@ def butterworth( ...@@ -13,10 +13,3 @@ def butterworth(
high_pass: bool = False, high_pass: bool = False,
squared_butterworth: bool = False, squared_butterworth: bool = False,
npad: Optional[int] = None) -> Matrix: ... npad: Optional[int] = None) -> Matrix: ...
def sosfreqz(sos: Any,
worN: Optional[Union[int, Vector]] = 512,
whole: bool = False,
fs: Optional[float] = None)\
-> tuple[Vector, Vector]: ...
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment