Fix tutorials

sdv-dev · Oct 25, 2023 · 39f7469 · 39f7469
1 parent 4e321a5
commit 39f7469
Show file tree

Hide file tree

Showing 5 changed files with 164,041 additions and 359 deletions.
diff --git a/copulas/visualization.py b/copulas/visualization.py
@@ -50,12 +50,13 @@ def _generate_1d_plot(data, title, labels, colors):
         title=title,
         plot_bgcolor=PlotConfig.BACKGROUND_COLOR,
         font={'size': PlotConfig.FONT_SIZE},
+        showlegend=True if labels[0] else False,
     )
 
     return fig
 
 
-def hist_1d(data, title=None, label=None):
+def dist_1d(data, title=None, label=None):
     """Plot the 1 dimensional data.
 
     Args:
@@ -72,7 +73,7 @@ def hist_1d(data, title=None, label=None):
     return _generate_1d_plot(
         data=[data],
         title=title,
-        labels=[label or 'Data'],
+        labels=[label],
         colors=[PlotConfig.DATACEBO_DARK]
     )
 
@@ -138,6 +139,7 @@ def _generate_scatter_2d_plot(data, columns, color_discrete_map, title):
         title=title,
         plot_bgcolor=PlotConfig.BACKGROUND_COLOR,
         font={'size': PlotConfig.FONT_SIZE},
+        showlegend=False if len(color_discrete_map) == 1 else True,
     )
 
     return fig
@@ -233,13 +235,16 @@ def _generate_scatter_3d_plot(data, columns, color_discrete_map, title):
         z=columns[2],
         color='Data',
         color_discrete_map=color_discrete_map,
-        symbol='Data'
+        symbol='Data',
     )
 
+    fig.update_traces(marker={'size': 5})
+
     fig.update_layout(
         title=title,
         plot_bgcolor=PlotConfig.BACKGROUND_COLOR,
         font={'size': PlotConfig.FONT_SIZE},
+        showlegend=False if len(color_discrete_map) == 1 else True,
     )
 
     return fig

diff --git a/tutorials/00_Quickstart.ipynb b/tutorials/00_Quickstart.ipynb
diff --git a/tutorials/01_Introduction_to_Copulas.ipynb b/tutorials/01_Introduction_to_Copulas.ipynb
diff --git a/tutorials/02_Univariate_Distributions.ipynb b/tutorials/02_Univariate_Distributions.ipynb
diff --git a/tutorials/03_Multivariate_Distributions.ipynb b/tutorials/03_Multivariate_Distributions.ipynb