Add condition to only plot non-highlighted paths in the case there ar…

…e some to plot
scikit-learn-contrib · Dec 21, 2018 · 8b070e8 · 8b070e8
1 parent f0ed2f2
commit 8b070e8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/stability_selection/stability_selection.py b/stability_selection/stability_selection.py
@@ -136,8 +136,9 @@ def plot_stability_path(stability_selection, threshold_highlight=None,
     x_grid = stability_selection.lambda_grid / np.max(stability_selection.lambda_grid)
 
     fig, ax = plt.subplots(1, 1, **kwargs)
-    ax.plot(x_grid, stability_selection.stability_scores_[~paths_to_highlight].T,
-            'k:', linewidth=0.5)
+    if not paths_to_highlight.all():
+        ax.plot(x_grid, stability_selection.stability_scores_[~paths_to_highlight].T,
+                'k:', linewidth=0.5)
 
     if paths_to_highlight.any():
         ax.plot(x_grid, stability_selection.stability_scores_[paths_to_highlight].T,