chore: type safety and refactoring

This commit is contained in:
jainanuj94
2024-08-05 23:57:24 +05:30
parent 7f0e5dd335
commit 103991746b

View File

@@ -1,6 +1,6 @@
import logging import logging
from pathlib import Path from pathlib import Path
from typing import List from typing import Dict, List
import joblib import joblib
import pandas as pd import pandas as pd
@@ -47,7 +47,7 @@ def _load_signal_candles(backtest_dir: Path):
return _load_backtest_analysis_data(backtest_dir, "signals") 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") return _load_backtest_analysis_data(backtest_dir, "exited")
@@ -71,8 +71,8 @@ def _process_candles_and_indicators(
def _analyze_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 buyf = signal_candles
if len(buyf) > 0: if len(buyf) > 0:
@@ -242,7 +242,7 @@ def _select_rows_by_tags(df, enter_reason_list, exit_reason_list):
def prepare_results( def prepare_results(
analysed_trades, stratname, enter_reason_list, exit_reason_list, timerange=None analysed_trades, stratname, enter_reason_list, exit_reason_list, timerange=None
): ) -> pd.DataFrame:
res_df = pd.DataFrame() res_df = pd.DataFrame()
for pair, trades in analysed_trades[stratname].items(): for pair, trades in analysed_trades[stratname].items():
if trades.shape[0] > 0: if trades.shape[0] > 0: