From 188f75d8ec312e6d2687fc2442ef2517abfbd62e Mon Sep 17 00:00:00 2001
From: robcaulk <rob.caulk@gmail.com>
Date: Sun, 18 Sep 2022 12:49:08 +0200
Subject: [PATCH] set model in models dict

---
 freqtrade/freqai/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/freqtrade/freqai/utils.py b/freqtrade/freqai/utils.py
index 50303019a..57bc17cdf 100644
--- a/freqtrade/freqai/utils.py
+++ b/freqtrade/freqai/utils.py
@@ -157,7 +157,7 @@ def plot_feature_importance(model: Any, pair: str, dk: FreqaiDataKitchen,
         for estimator, label in zip(model.estimators_, dk.label_list):
             models[label] = estimator
     else:
-        models[dk.label_list[0]]
+        models[dk.label_list[0]] = model
 
     for label in models:
         mdl = models[label]