From 5406404bde1d5199c7e75df9434edd7a3c74fff9 Mon Sep 17 00:00:00 2001
From: Felix Fischer <f.fischer@ifas.rwth-aachen.de>
Date: Wed, 26 Jun 2024 16:19:38 +0200
Subject: [PATCH] Made typing for function calls more specific

---
 src/scipy-stubs/signal/__init__.pyi | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/scipy-stubs/signal/__init__.pyi b/src/scipy-stubs/signal/__init__.pyi
index 4592fb8..7046d28 100644
--- a/src/scipy-stubs/signal/__init__.pyi
+++ b/src/scipy-stubs/signal/__init__.pyi
@@ -8,6 +8,9 @@ from typing import Tuple, Optional, Union, overload, TypeVar
 from scientific_plots.types_ import Vector, Matrix, Tensor
 
 
+Input = TypeVar("Input", Vector, Matrix, Tensor)
+
+
 def welch(
     Y: Union[Vector, list[float]],
     fs: Optional[float] = None,
@@ -47,17 +50,17 @@ def savgol_filter(
 
 
 def sosfilt(
-    sos: Single, x: Union[Single, Matrix],
+    sos: Single, x: Input,
     axis: int = -1, zi: Optional[Single] = None,
     padtype: Optional[str] = "odd", padlen: Optional[int] = None)\
-        -> Vector: ...
+        -> Input: ...
 
 
 def sosfiltfilt(
-    sos: Single, x: Union[Single, Matrix],
+    sos: Single, x: Input,
     axis: int = -1, zi: Optional[Single] = None,
     padtype: Optional[str] = "odd", padlen: Optional[int] = None)\
-        -> Vector: ...
+        -> Input: ...
 
 
 def butter(
@@ -65,9 +68,6 @@ def butter(
     output: str = "ba", fs: Optional[float] = None) -> Vector: ...
 
 
-Input = TypeVar("Input", Vector, Matrix, Tensor)
-
-
 def wiener(
     im: Input,
     noise: Optional[float] = None,
-- 
GitLab