ensure model_type is defined

2022-11-13 18:50:25 +01:00 · 2022-11-13 18:50:25 +01:00 · bf4d5b432a
commit bf4d5b432a
parent 6394ef4558
1 changed files with 1 additions and 0 deletions
--- a/freqtrade/freqai/data_drawer.py
+++ b/freqtrade/freqai/data_drawer.py
@ -99,6 +99,7 @@ class FreqaiDataDrawer:
                "model_filename": "", "trained_timestamp": 0,
                "data_path": "", "extras": {}}
        if 'Reinforcement' in self.config['freqaimodel']:
+            self.model_type = 'stable_baselines'
            logger.warning('User passed a ReinforcementLearner model, FreqAI will '
                           'now use stable_baselines3 to save models.')
        else: