Avoid list / int division error in relimp

Fix #399
raphaelvallat · Jan 29, 2024 · d26af45 · d26af45
1 parent ae09514
commit d26af45
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pingouin/regression.py b/pingouin/regression.py
@@ -593,7 +593,7 @@ def _relimp(S):
     stats_relimp = {
         "names": predictors,
         "relimp": all_preds,
-        "relimp_perc": all_preds / sum(all_preds) * 100,
+        "relimp_perc": np.array(all_preds) / sum(all_preds) * 100,
     }
 
     return stats_relimp