seaborn API fix

ondrolexa · Feb 16, 2024 · 16a48be · 16a48be
1 parent 9aa96fb
commit 16a48be
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 22 deletions.
diff --git a/ci-environment.yml b/ci-environment.yml
diff --git a/polylx/core.py b/polylx/core.py
@@ -2134,11 +2134,11 @@ def _seaborn_plot(self, sns_plot_fun, val, **kwargs):
 
         Args:
             sns_plot_fun: sns_plotting function
-            val: array of values
+            attr: object attribute to be used. See .df method
 
         Keyword Args:
           show: If True matplotlib show is called. Default True
-          attr: property used for plotting.
+          ax: None or matplotlib axes to be used
           hue: When True attr is used for hue and names for x.
 
         When show=False, returns matplotlib axes object.
@@ -2166,6 +2166,7 @@ def _seaborn_plot(self, sns_plot_fun, val, **kwargs):
                 x="class",
                 y=val,
                 data=self.df("class", val),
+                hue="class",
                 order=self.classes.index,
                 palette=self.classes._colors_dict,
             )
@@ -2211,6 +2212,7 @@ def countplot(self, **kwargs):
             sns.countplot(
                 x="class",
                 data=self.df("class"),
+                hue="class",
                 order=self.classes.index,
                 palette=self.classes._colors_dict,
             )