X-axis for the bar plot should be labeled Value instead of `Categor…

…y` (#621)
sdv-dev · Aug 12, 2024 · 6491ccf · 6491ccf
1 parent 685731f
commit 6491ccf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sdmetrics/visualization.py b/sdmetrics/visualization.py
@@ -254,7 +254,7 @@ def _generate_column_plot(
         plot_title (str, optional):
             Title to use for the plot. Defaults to 'Real vs. Synthetic Data for column {column}'
         x_label (str, optional):
-            Label to use for x-axis. Defaults to 'Category'.
+            Label to use for x-axis. Defaults to 'Value'.
 
     Returns:
         plotly.graph_objects._figure.Figure
@@ -353,7 +353,7 @@ def _generate_column_plot(
         plot_title = title
 
     if not x_label:
-        x_label = 'Category'
+        x_label = 'Value'
 
     fig.update_layout(
         title=plot_title,

diff --git a/tests/unit/test_visualization.py b/tests/unit/test_visualization.py
@@ -506,7 +506,7 @@ def test___generate_column_plot_type_bar(mock_bar_plot):
     mock_bar_plot.assert_called_once_with(ANY, ANY, {})
     mock_fig.update_layout.assert_called_once_with(
         title="Real vs. Synthetic Data for column 'values'",
-        xaxis_title='Category',
+        xaxis_title='Value',
         yaxis_title='Frequency',
         plot_bgcolor=PlotConfig.BACKGROUND_COLOR,
         annotations=[],