feat: ✨ return the figure

ThomasBury · Apr 29, 2024 · 33469b7 · 33469b7
1 parent 6f550c8
commit 33469b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/arfs/feature_selection/variable_importance.py b/src/arfs/feature_selection/variable_importance.py
@@ -313,7 +313,7 @@ def plot_importance(
         ax2.set(frame_on=False)
 
         fig.align_labels()
-        plt.show()
+        return fig
 
 
 def _compute_varimp_lgb(