Simplify import if only one element is used

This commit is contained in:
Matthias
2023-06-17 10:12:36 +02:00
parent 34e7e3efea
commit 6bb75f0dd4
+9 -9
View File
@@ -5,7 +5,7 @@ from datetime import datetime, timedelta, timezone
from pathlib import Path from pathlib import Path
from typing import Any, Dict, List, Optional from typing import Any, Dict, List, Optional
import pandas as pd from pandas import DataFrame
from freqtrade.configuration import TimeRange from freqtrade.configuration import TimeRange
from freqtrade.data.history import get_timerange from freqtrade.data.history import get_timerange
@@ -20,10 +20,10 @@ logger = logging.getLogger(__name__)
class VarHolder: class VarHolder:
timerange: TimeRange timerange: TimeRange
data: pd.DataFrame data: DataFrame
indicators: pd.DataFrame indicators: Dict[str, DataFrame]
result: pd.DataFrame result: DataFrame
compared: pd.DataFrame compared: DataFrame
from_dt: datetime from_dt: datetime
to_dt: datetime to_dt: datetime
compared_dt: datetime compared_dt: datetime
@@ -63,7 +63,7 @@ class LookaheadAnalysis:
return timestamp return timestamp
@staticmethod @staticmethod
def get_result(backtesting: Backtesting, processed: pd.DataFrame): def get_result(backtesting: Backtesting, processed: DataFrame):
min_date, max_date = get_timerange(processed) min_date, max_date = get_timerange(processed)
result = backtesting.backtest( result = backtesting.backtest(
@@ -92,8 +92,8 @@ class LookaheadAnalysis:
# analyzes two data frames with processed indicators and shows differences between them. # analyzes two data frames with processed indicators and shows differences between them.
def analyze_indicators(self, full_vars: VarHolder, cut_vars: VarHolder, current_pair): def analyze_indicators(self, full_vars: VarHolder, cut_vars: VarHolder, current_pair):
# extract dataframes # extract dataframes
cut_df = cut_vars.indicators[current_pair] cut_df: DataFrame = cut_vars.indicators[current_pair]
full_df = full_vars.indicators[current_pair] full_df: DataFrame = full_vars.indicators[current_pair]
# cut longer dataframe to length of the shorter # cut longer dataframe to length of the shorter
full_df_cut = full_df[ full_df_cut = full_df[
@@ -127,7 +127,7 @@ class LookaheadAnalysis:
f"{col_name[0]}. " f"{col_name[0]}. "
f"{str(self_value)} != {str(other_value)}") f"{str(self_value)} != {str(other_value)}")
def prepare_data(self, varholder: VarHolder, pairs_to_load: List[pd.DataFrame]): def prepare_data(self, varholder: VarHolder, pairs_to_load: List[DataFrame]):
if 'freqai' in self.local_config and 'identifier' in self.local_config['freqai']: if 'freqai' in self.local_config and 'identifier' in self.local_config['freqai']:
# purge previous data if the freqai model is defined # purge previous data if the freqai model is defined