Bugfix - Matplotlib - Candlestick Plot (#779)

* Bugfix - Matplotlib - Candlestick Plot Color option was not working for list inputs. * black fix
vertica · Oct 27, 2023 · 78743cc · 78743cc
1 parent d493c43
commit 78743cc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/verticapy/machine_learning/memmodel/naive_bayes.py b/verticapy/machine_learning/memmodel/naive_bayes.py
@@ -84,7 +84,7 @@ class NaiveBayes(MulticlassClassifier):
                 'male': 0.583333333333333},
                 'S': {'female': 0.311212814645309,
                 'male': 0.688787185354691}}
-                
+
         prior: ArrayLike
             The model's classes probabilities.
         classes: ArrayLike

diff --git a/verticapy/plotting/_matplotlib/candlestick.py b/verticapy/plotting/_matplotlib/candlestick.py
@@ -61,7 +61,7 @@ def draw(
         if "color" in style_kwargs:
             color_list = (
                 style_kwargs["color"] + color_list
-                if isinstance(style_kwargs, list)
+                if isinstance(style_kwargs["color"], list)
                 else [style_kwargs["color"]] + color_list
             )
             style_kwargs.pop("color")