From bf4d5b432a19a090c1e09eb499c92d56258a7004 Mon Sep 17 00:00:00 2001
From: robcaulk <rob.caulk@gmail.com>
Date: Sun, 13 Nov 2022 18:50:25 +0100
Subject: [PATCH] ensure model_type is defined

---
 freqtrade/freqai/data_drawer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/freqtrade/freqai/data_drawer.py b/freqtrade/freqai/data_drawer.py
index 590439697..96b481074 100644
--- 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: