Merge pull request #17 from ibr-cm/fix_count_plot

plots: fix count plot axis parameter
ibr-cm · Aug 20, 2024 · 62490de · 62490de
2 parents c96a3f7 + 31812f5
commit 62490de
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/plots.py b/plots.py
@@ -876,12 +876,20 @@ def plot_catplot(self, df, x='v2x_rate', y='cbr', hue='moduleName', row='dcc', c
         kwargs = self.set_plot_specific_options(plot_type, **kwargs)
 
         logd(f'PlottingTask::plot_catplot: {df.columns=}')
-        grid = sb.catplot(data=df, x=x, y=y, row=row, col=column
-                        , hue=hue
-                        , kind=plot_type
-                        # , legend_out=False
-                        , **kwargs
-                       )
+        if plot_type != 'count':
+            grid = sb.catplot(data=df, x=x, y=y, row=row, col=column
+                            , hue=hue
+                            , kind=plot_type
+                            # , legend_out=False
+                            , **kwargs
+                           )
+        else:
+            grid = sb.catplot(data=df, x=x, row=row, col=column
+                            , hue=hue
+                            , kind=plot_type
+                            # , legend_out=False
+                            , **kwargs
+                           )
 
         grid = self.set_grid_defaults(grid)