From 6423b59670aed012b4a4c0ba38f0f5b07d044607 Mon Sep 17 00:00:00 2001 From: csrgxtu Date: Thu, 1 Aug 2024 19:37:34 +0800 Subject: [PATCH] fix seaborn-whitegrid not found when use visualization.plot --- src/adtk/visualization/_visualization.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/adtk/visualization/_visualization.py b/src/adtk/visualization/_visualization.py index 92c0cce..a5f09b0 100644 --- a/src/adtk/visualization/_visualization.py +++ b/src/adtk/visualization/_visualization.py @@ -198,7 +198,10 @@ def plot( """ # setup style - plt.style.use("seaborn-whitegrid") + if "seaborn-whitegrid" in plt.style.available: + plt.style.use("seaborn-whitegrid") + elif plt.style.available: + plt.style.use(plt.style.available[0]) # initialize color generator color_generator = ColorGenerator()