From 90efd046179f4281e61b4f7ee2fe3b91a78161cb Mon Sep 17 00:00:00 2001 From: Matthias Date: Tue, 11 Jun 2024 06:47:23 +0200 Subject: [PATCH] Improve typing in backtesting --- freqtrade/optimize/optimize_reports/optimize_reports.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/freqtrade/optimize/optimize_reports/optimize_reports.py b/freqtrade/optimize/optimize_reports/optimize_reports.py index 2ca467eb6..abb831d36 100644 --- a/freqtrade/optimize/optimize_reports/optimize_reports.py +++ b/freqtrade/optimize/optimize_reports/optimize_reports.py @@ -1,7 +1,7 @@ import logging from copy import deepcopy from datetime import datetime, timedelta, timezone -from typing import Any, Dict, List, Tuple, Union +from typing import Any, Dict, List, Literal, Tuple, Union import numpy as np from pandas import DataFrame, Series, concat, to_datetime @@ -141,7 +141,10 @@ def generate_pair_metrics( def generate_tag_metrics( - tag_type: str, starting_balance: int, results: DataFrame, skip_nan: bool = False + tag_type: Literal["enter_tag", "exit_reason"], + starting_balance: int, + results: DataFrame, + skip_nan: bool = False, ) -> List[Dict]: """ Generates and returns a list of metrics for the given tag trades and the results dataframe