diff --git a/elastalert/test_rule.py b/elastalert/test_rule.py index 9bb16560..951afe14 100644 --- a/elastalert/test_rule.py +++ b/elastalert/test_rule.py @@ -137,7 +137,7 @@ def _parse_args(self, args: list) -> argparse.Namespace: return parsed_args - def str_to_ts(self, input: str) -> datetime: + def str_to_ts(self, input: str) -> datetime.datetime: if input == "NOW": return self.ts_now @@ -148,7 +148,7 @@ def str_to_ts(self, input: str) -> datetime: f"Input is not a valid ISO8601 timestamp (YYYY-MM-DDTHH:MM:SS+XX:00): {input}" ) - def parse_starttime(self, timeframe=None) -> datetime: + def parse_starttime(self, timeframe=None) -> datetime.datetime: if self.args.start: try: return self.str_to_ts(self.args.start)