From 0a4f82bc22794a33cf0047741bd3a1ed10c9d578 Mon Sep 17 00:00:00 2001 From: James Lin Date: Sat, 26 Apr 2025 10:24:53 -0400 Subject: [PATCH] update columns from ORDERFLOW_ADDED_COLUMNS --- freqtrade/freqai/data_kitchen.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/freqtrade/freqai/data_kitchen.py b/freqtrade/freqai/data_kitchen.py index eac1426fb..4b49c3495 100644 --- a/freqtrade/freqai/data_kitchen.py +++ b/freqtrade/freqai/data_kitchen.py @@ -17,7 +17,7 @@ from sklearn.model_selection import train_test_split from freqtrade.configuration import TimeRange from freqtrade.constants import DOCS_LINK, Config -from freqtrade.data.converter import reduce_dataframe_footprint +from freqtrade.data.converter import ORDERFLOW_ADDED_COLUMNS, reduce_dataframe_footprint from freqtrade.exceptions import OperationalException from freqtrade.exchange import timeframe_to_seconds from freqtrade.strategy import merge_informative_pair @@ -709,11 +709,7 @@ class FreqaiDataKitchen: skip_columns = [ (f"{s}_{suffix}") for s in ["date", "open", "high", "low", "close", "volume"] ] - for s in [ - 'trades', 'orderflow', 'imbalances', 'stacked_imbalances_bid', - 'stacked_imbalances_ask', 'max_delta', 'min_delta', 'bid', 'ask', - 'delta', 'total_trades' - ]: + for s in ORDERFLOW_ADDED_COLUMNS: if s in dataframe.columns and f"{s}_{suffix}" in dataframe.columns: skip_columns.append(f"{s}_{suffix}") dataframe = dataframe.drop(columns=skip_columns)