diff --git a/saqc/core/core.py b/saqc/core/core.py index 1a1976fea27b6e9a4e503bf84ccd79b1a4b7a68e..916bcf3e48bcb984fe6cba28b5270297130bef33 100644 --- a/saqc/core/core.py +++ b/saqc/core/core.py @@ -54,16 +54,20 @@ class SaQC(FunctionsMixin): def __init__( self, - data: pd.Series - | pd.DataFrame - | DictOfSeries - | list[pd.Series | pd.DataFrame | DictOfSeries] - | None = None, - flags: pd.DataFrame - | DictOfSeries - | Flags - | list[pd.DataFrame | DictOfSeries | Flags] - | None = None, + data: ( + pd.Series + | pd.DataFrame + | DictOfSeries + | list[pd.Series | pd.DataFrame | DictOfSeries] + | None + ) = None, + flags: ( + pd.DataFrame + | DictOfSeries + | Flags + | list[pd.DataFrame | DictOfSeries | Flags] + | None + ) = None, scheme: str | TranslationScheme = "float", ): self.scheme: TranslationScheme = scheme @@ -193,12 +197,14 @@ class SaQC(FunctionsMixin): def __setitem__( self, key: str | slice | Iterable[str], - value: SaQC - | pd.Series - | pd.DataFrame - | DictOfSeries - | dict[Any, pd.Series] - | Iterable[pd.Series], + value: ( + SaQC + | pd.Series + | pd.DataFrame + | DictOfSeries + | dict[Any, pd.Series] + | Iterable[pd.Series] + ), ): keys = self._get_keys(key) if isinstance(value, SaQC): diff --git a/saqc/core/flags.py b/saqc/core/flags.py index 4f1d83df7fa210b7c2597174dbe5d20dda30438c..c90d05b6e2d58c66871dab2389c0d61b2a472162 100644 --- a/saqc/core/flags.py +++ b/saqc/core/flags.py @@ -323,12 +323,10 @@ class Flags: # item access @overload - def __getitem__(self, key: str) -> pd.Series: - ... + def __getitem__(self, key: str) -> pd.Series: ... @overload - def __getitem__(self, key: list | pd.Index) -> Flags: - ... + def __getitem__(self, key: list | pd.Index) -> Flags: ... def __getitem__(self, key: str | list | pd.Index) -> pd.Series | Flags: if isinstance(key, str): diff --git a/saqc/core/translation/basescheme.py b/saqc/core/translation/basescheme.py index 92aceb333017222b9700c56a65db36d0695d8e31..c072907c8ddb477ccd8e112d23822d4414212192 100644 --- a/saqc/core/translation/basescheme.py +++ b/saqc/core/translation/basescheme.py @@ -217,7 +217,6 @@ class MappingScheme(TranslationScheme): class FloatScheme(TranslationScheme): - """ Acts as the default Translator, provides a changeable subset of the internal float flags diff --git a/saqc/core/translation/dmpscheme.py b/saqc/core/translation/dmpscheme.py index c7708e42d56f81b4b83cc845d26b1ab062f851d8..6ecd324d017847fb13cc6add3d2c41a7264d7849 100644 --- a/saqc/core/translation/dmpscheme.py +++ b/saqc/core/translation/dmpscheme.py @@ -41,7 +41,6 @@ _QUALITY_LABELS = [ class DmpScheme(MappingScheme): - """ Implements the translation from and to the flagging scheme implemented in the UFZ - Datamanagementportal diff --git a/saqc/core/translation/positionalscheme.py b/saqc/core/translation/positionalscheme.py index fffc988341204fd21d2136d7db5f3a1990de0bf2..6e4bbe4833663d14dd3b50927d0ed2ab6c00d821 100644 --- a/saqc/core/translation/positionalscheme.py +++ b/saqc/core/translation/positionalscheme.py @@ -16,7 +16,6 @@ from saqc.core.translation.basescheme import BackwardMap, ForwardMap, MappingSch class PositionalScheme(MappingScheme): - """ Implements the translation from and to the flagging scheme implemented by CHS """ diff --git a/saqc/core/translation/simplescheme.py b/saqc/core/translation/simplescheme.py index aabd0472b55817696156ec156d036484a7f7264b..1ee136e19a0d4dad0c6214c63d1aa4d64bf19c68 100644 --- a/saqc/core/translation/simplescheme.py +++ b/saqc/core/translation/simplescheme.py @@ -10,7 +10,6 @@ from saqc.core.translation import MappingScheme class SimpleScheme(MappingScheme): - """ Acts as the default Translator, provides a changeable subset of the internal float flags diff --git a/saqc/funcs/generic.py b/saqc/funcs/generic.py index b03924585f9db6519461dcfd90a8fff62310e97a..94fa02eea9796a6e3d3891821d282d2fbc8bf751 100644 --- a/saqc/funcs/generic.py +++ b/saqc/funcs/generic.py @@ -25,8 +25,9 @@ class GenericFunction(Protocol): __name__: str __globals__: dict[str, Any] - def __call__(self, *args: pd.Series) -> pd.Series | pd.DataFrame | DictOfSeries: - ... # pragma: no cover + def __call__( + self, *args: pd.Series + ) -> pd.Series | pd.DataFrame | DictOfSeries: ... # pragma: no cover def _flagSelect(field: str, flags: Flags, label: str | None = None) -> pd.Series: diff --git a/saqc/funcs/noise.py b/saqc/funcs/noise.py index cde2b487c0e14dfbc1823beef6b2c1975280e5f1..18edfd7b1b3c16aef4dd970e6cab9aac3c706b89 100644 --- a/saqc/funcs/noise.py +++ b/saqc/funcs/noise.py @@ -36,8 +36,9 @@ class NoiseMixin: field: str, window: str | pd.Timedelta, thresh: float, - func: Literal["std", "var", "mad"] - | Callable[[np.ndarray, pd.Series], float] = "std", + func: ( + Literal["std", "var", "mad"] | Callable[[np.ndarray, pd.Series], float] + ) = "std", sub_window: str | pd.Timedelta | None = None, sub_thresh: float | None = None, min_periods: int | None = None, @@ -102,8 +103,9 @@ class NoiseMixin: field: str, window: str | pd.Timedelta, thresh: float, - func: Literal["std", "var", "mad"] - | Callable[[np.ndarray, pd.Series], float] = "std", + func: ( + Literal["std", "var", "mad"] | Callable[[np.ndarray, pd.Series], float] + ) = "std", sub_window: str | pd.Timedelta | None = None, sub_thresh: float | None = None, min_periods: int | None = None, diff --git a/saqc/lib/types.py b/saqc/lib/types.py index d9acc44660eeafbaf32adf76ba3bedf640f27c71..8b5b6e5fa77b1e1c0f443f9bc50abdb694362808 100644 --- a/saqc/lib/types.py +++ b/saqc/lib/types.py @@ -30,8 +30,9 @@ ExternalFlag = Union[str, float, int] # needed for deeper type hinting magic class CurveFitter(Protocol): - def __call__(self, data: np.ndarray, *params: float) -> np.ndarray: - ... # pragma: no cover + def __call__( + self, data: np.ndarray, *params: float + ) -> np.ndarray: ... # pragma: no cover class Comparable(Protocol):