From e229902381ae21107cf379170ca92acb0cc841a2 Mon Sep 17 00:00:00 2001 From: robcaulk Date: Mon, 30 May 2022 12:48:22 +0200 Subject: [PATCH] fix bug in previous commit --- freqtrade/freqai/data_drawer.py | 5 +++-- freqtrade/freqai/freqai_interface.py | 4 ++-- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/freqtrade/freqai/data_drawer.py b/freqtrade/freqai/data_drawer.py index 6a5393cb8..5421db6cf 100644 --- a/freqtrade/freqai/data_drawer.py +++ b/freqtrade/freqai/data_drawer.py @@ -86,14 +86,15 @@ class FreqaiDataDrawer: # send pair to end of queue self.pair_dict[pair]['priority'] = len(self.pair_dict) - def set_initial_return_values(self, pair, dh): + def set_initial_return_values(self, pair: str, dh): + self.model_return_values[pair] = {} self.model_return_values[pair]['predictions'] = dh.full_predictions self.model_return_values[pair]['do_preds'] = dh.full_do_predict self.model_return_values[pair]['target_mean'] = dh.full_target_mean self.model_return_values[pair]['target_std'] = dh.full_target_std - def append_model_predictions(self, pair, predictions, do_preds, + def append_model_predictions(self, pair: str, predictions, do_preds, target_mean, target_std, dh) -> None: pred_store = self.model_return_values[pair]['predictions'] diff --git a/freqtrade/freqai/freqai_interface.py b/freqtrade/freqai/freqai_interface.py index 386fab9fc..014b80208 100644 --- a/freqtrade/freqai/freqai_interface.py +++ b/freqtrade/freqai/freqai_interface.py @@ -227,8 +227,8 @@ class IFreqaiModel(ABC): else: preds, do_preds = self.predict(dataframe.iloc[-2:], dh) self.data_drawer.append_model_predictions(metadata['pair'], preds, do_preds, - self.dh.data["target_mean"], - self.dh.data["target_std"], dh) + dh.data["target_mean"], + dh.data["target_std"], dh) return dh