Update MAM.py

styles mistake correction
DP6 · Nov 25, 2024 · 03a836e · 03a836e
1 parent 72933d9
commit 03a836e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/marketing_attribution_models/MAM.py b/marketing_attribution_models/MAM.py
@@ -9,7 +9,10 @@
 import pandas as pd
 import seaborn as sns
 
-plt.style.use("seaborn-white")
+if "seaborn-white" in plt.style.available:
+    plt.style.use("seaborn-white")
+else:
+    plt.style.use("default")
 
 
 class MAM: