diff --git a/src/data_morph/data/loader.py b/src/data_morph/data/loader.py index dfd62f7b..4a3e49de 100644 --- a/src/data_morph/data/loader.py +++ b/src/data_morph/data/loader.py @@ -136,14 +136,15 @@ def plot_available_datasets(cls) -> Axes: The list of available datasets built into Data Morph. """ num_plots = len(cls.AVAILABLE_DATASETS) - num_cols = 3 + num_cols = 4 num_rows = int(np.ceil(num_plots / num_cols)) + scale = 4 fig, axs = plt.subplots( num_rows, num_cols, layout='constrained', - figsize=(12, 4 * num_rows), + figsize=(scale * num_cols, scale * num_rows), subplot_kw={'aspect': 'equal'}, ) fig.get_layout_engine().set(w_pad=0.2, h_pad=0.2) diff --git a/tests/data/test_loader.py b/tests/data/test_loader.py index ecb66532..a1a5d1bb 100644 --- a/tests/data/test_loader.py +++ b/tests/data/test_loader.py @@ -73,3 +73,4 @@ def test_plot_available_datasets(self, monkeypatch, subset): assert subplot_title.endswith(' points)') if dataset in ['Python', 'SDS']: assert 'logo' in subplot_title + assert ax.get_aspect() == 1