From 103991746bfe29725a1634f1157c9b344e0b23a7 Mon Sep 17 00:00:00 2001 From: jainanuj94 Date: Mon, 5 Aug 2024 23:57:24 +0530 Subject: [PATCH] chore: type safety and refactoring --- freqtrade/data/entryexitanalysis.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/freqtrade/data/entryexitanalysis.py b/freqtrade/data/entryexitanalysis.py index f2440a787..99b0e3bcb 100644 --- a/freqtrade/data/entryexitanalysis.py +++ b/freqtrade/data/entryexitanalysis.py @@ -1,6 +1,6 @@ import logging from pathlib import Path -from typing import List +from typing import Dict, List import joblib import pandas as pd @@ -47,7 +47,7 @@ def _load_signal_candles(backtest_dir: Path): return _load_backtest_analysis_data(backtest_dir, "signals") -def _load_exit_signal_candles(backtest_dir: Path): +def _load_exit_signal_candles(backtest_dir: Path) -> Dict[str, Dict[str, pd.DataFrame]]: return _load_backtest_analysis_data(backtest_dir, "exited") @@ -71,8 +71,8 @@ def _process_candles_and_indicators( def _analyze_candles_and_indicators( - pair, trades: pd.DataFrame, signal_candles: pd.DataFrame, analyse_on="open_date" -): + pair: str, trades: pd.DataFrame, signal_candles: pd.DataFrame, analyse_on="open_date" +) -> pd.DataFrame: buyf = signal_candles if len(buyf) > 0: @@ -242,7 +242,7 @@ def _select_rows_by_tags(df, enter_reason_list, exit_reason_list): def prepare_results( analysed_trades, stratname, enter_reason_list, exit_reason_list, timerange=None -): +) -> pd.DataFrame: res_df = pd.DataFrame() for pair, trades in analysed_trades[stratname].items(): if trades.shape[0] > 0: