From 68f7a315048703285e978dc58e023506153069f2 Mon Sep 17 00:00:00 2001
From: robcaulk <rob.caulk@gmail.com>
Date: Sun, 18 Sep 2022 00:00:14 +0200
Subject: [PATCH] ensure continued operation despite not being able to plot

---
 freqtrade/freqai/utils.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/freqtrade/freqai/utils.py b/freqtrade/freqai/utils.py
index 3f278d436..10afbaf52 100644
--- a/freqtrade/freqai/utils.py
+++ b/freqtrade/freqai/utils.py
@@ -156,6 +156,8 @@ def plot_feature_importance(model: Any, pair: str, dk: FreqaiDataKitchen,
     if 'FreqaiMultiOutputRegressor' in str(model.__class__):
         for estimator, label in zip(model.estimators_, dk.label_list):
             models[label] = estimator
+    else:
+        models[dk.label_list[0]]
 
     for label in models:
         mdl = models[label]
@@ -164,8 +166,8 @@ def plot_feature_importance(model: Any, pair: str, dk: FreqaiDataKitchen,
         elif "lightgbm.sklearn" or "xgb" in str(mdl.__class__):
             feature_importance = mdl.feature_importances_
         else:
-            # TODO: Add support for more libraries
-            raise NotImplementedError(f"Cannot extract feature importance from {mdl.__class__}")
+            logger.info('Model type not support for generating feature importances.')
+            return
 
         # Data preparation
         fi_df = pd.DataFrame({