BUILD make matplotlib an optional import

DataResponsibly · Dec 12, 2023 · 8f0eeb8 · 8f0eeb8
1 parent 8260c3c
commit 8f0eeb8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/sharp/visualization/_waterfall.py b/sharp/visualization/_waterfall.py
@@ -1,8 +1,7 @@
 import re
 import pandas as pd
 import numpy as np
-import matplotlib
-import matplotlib.pyplot as plt
+from sharp.utils._utils import _optional_import
 
 # from sharp.utils import scores_to_rank
 
@@ -25,6 +24,9 @@ def format_value(s, format_str):
 
 
 def _waterfall(shap_values, max_display=10, show=False):  # noqa
+    matplotlib = _optional_import("matplotlib")
+    plt = _optional_import("matplotlib.pyplot")
+
     # Turn off interactive plot
     if show is False:
         plt.ioff()