diff --git a/marketing_attribution_models/MAM.py b/marketing_attribution_models/MAM.py index 9323671..b918928 100644 --- 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: