feat: align colorbar for categorical color column

datamole-ai · Nov 1, 2023 · 493fee3 · 493fee3
1 parent 2502938
commit 493fee3
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/edvart/report_sections/multivariate_analysis.py b/edvart/report_sections/multivariate_analysis.py
@@ -647,6 +647,13 @@ def parallel_coordinates(
                     "len": min(40 * len(categories), 300),
                 }
             )
+            # Align the colorbar with the categories
+            line.update(
+                {
+                    "cmin": -0.5,
+                    "cmax": len(categories) - 0.5,
+                }
+            )
     else:
         line = None
 
@@ -817,6 +824,13 @@ def parallel_categories(
                     "len": min(40 * len(categories), 300),
                 }
             )
+            # Align the colorbar with the categories
+            line.update(
+                {
+                    "cmin": -0.5,
+                    "cmax": len(categories) - 0.5,
+                }
+            )
     else:
         line = None