Skip to content

Commit 6dccab9

Browse files
committed
refactor: simplify backtest analysis data loading
1 parent 653c160 commit 6dccab9

File tree

3 files changed

+6
-30
lines changed

3 files changed

+6
-30
lines changed

freqtrade/data/btanalysis/__init__.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,7 @@
1616
load_backtest_data,
1717
load_backtest_metadata,
1818
load_backtest_stats,
19-
load_exit_signal_candles,
2019
load_file_from_zip,
21-
load_rejected_signals,
22-
load_signal_candles,
2320
load_trades,
2421
load_trades_from_db,
2522
trade_list_to_dataframe,

freqtrade/data/btanalysis/bt_fileutils.py

Lines changed: 0 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -473,27 +473,6 @@ def load_backtest_analysis_data(backtest_dir: Path, name: Literal["signals", "re
473473
return None
474474

475475

476-
def load_rejected_signals(backtest_dir: Path):
477-
"""
478-
Load rejected signals from backtest directory
479-
"""
480-
return load_backtest_analysis_data(backtest_dir, "rejected")
481-
482-
483-
def load_signal_candles(backtest_dir: Path):
484-
"""
485-
Load signal candles from backtest directory
486-
"""
487-
return load_backtest_analysis_data(backtest_dir, "signals")
488-
489-
490-
def load_exit_signal_candles(backtest_dir: Path) -> dict[str, dict[str, pd.DataFrame]]:
491-
"""
492-
Load exit signal candles from backtest directory
493-
"""
494-
return load_backtest_analysis_data(backtest_dir, "exited")
495-
496-
497476
def trade_list_to_dataframe(trades: list[Trade] | list[LocalTrade]) -> pd.DataFrame:
498477
"""
499478
Convert list of Trade objects to pandas Dataframe

freqtrade/data/entryexitanalysis.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -7,11 +7,9 @@
77
from freqtrade.constants import Config
88
from freqtrade.data.btanalysis import (
99
BT_DATA_COLUMNS,
10+
load_backtest_analysis_data,
1011
load_backtest_data,
1112
load_backtest_stats,
12-
load_exit_signal_candles,
13-
load_rejected_signals,
14-
load_signal_candles,
1513
)
1614
from freqtrade.exceptions import ConfigurationError, OperationalException
1715
from freqtrade.util import print_df_rich_table
@@ -354,12 +352,14 @@ def process_entry_exit_reasons(config: Config):
354352
trades = load_backtest_data(config["exportfilename"], strategy_name)
355353

356354
if trades is not None and not trades.empty:
357-
signal_candles = load_signal_candles(config["exportfilename"])
358-
exit_signals = load_exit_signal_candles(config["exportfilename"])
355+
signal_candles = load_backtest_analysis_data(config["exportfilename"], "signals")
356+
exit_signals = load_backtest_analysis_data(config["exportfilename"], "exited")
359357

360358
rej_df = None
361359
if do_rejected:
362-
rejected_signals_dict = load_rejected_signals(config["exportfilename"])
360+
rejected_signals_dict = load_backtest_analysis_data(
361+
config["exportfilename"], "rejected"
362+
)
363363
rej_df = prepare_results(
364364
rejected_signals_dict,
365365
strategy_name,

0 commit comments

Comments
 (0)