From d2a65c8757ddf42eac1cb6a09ab64df575afa229 Mon Sep 17 00:00:00 2001
From: David Schaefer <david.schaefer@ufz.de>
Date: Mon, 21 Nov 2022 14:09:22 +0100
Subject: [PATCH] type hints
---
saqc/core/flags.py | 2 +-
saqc/core/translation/basescheme.py | 4 ++--
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/saqc/core/flags.py b/saqc/core/flags.py
index 48482a703..4f6424a26 100644
--- a/saqc/core/flags.py
+++ b/saqc/core/flags.py
@@ -192,7 +192,7 @@ class Flags:
"""
def __init__(
- self, raw_data: Optional[Union[DictLike, Flags]] = None, copy: bool = False
+ self, raw_data: DictLike | Flags | None = None, copy: bool = False
):
self._data: dict[str, History]
diff --git a/saqc/core/translation/basescheme.py b/saqc/core/translation/basescheme.py
index 511fc92d0..78347b020 100644
--- a/saqc/core/translation/basescheme.py
+++ b/saqc/core/translation/basescheme.py
@@ -22,14 +22,14 @@ from saqc.lib.types import ExternalFlag, PandasLike
ForwardMap = Dict[ExternalFlag, float]
BackwardMap = Dict[float, ExternalFlag]
-class TranslationScheme:
+class TranslationScheme:
@abstractmethod
def __call__(self, ExternalFlag) -> float:
pass
@abstractmethod
- def forward(self, PandasLike) -> Flags:
+ def forward(self, pd.DataFrame | DictOfSeries) -> Flags:
pass
@abstractmethod
--
GitLab