Unable to create a heatmap with only the real data

[npatki]

Environment Details

• SDMetrics version: 0.16.0

Error Description

According to this documentation I should be able to use the get_column_pair_plot function to to visualize just the real data (if I don't have access to the synthetic data).

This function currently works with plot types scatter and box but it does not work with heatmap.

Steps to reproduce

from sdv.datasets.demo import get_available_demos, download_demo
from sdv.evaluation.single_table import get_column_pair_plot

data, metadata = download_demo(
    modality='single_table',
    dataset_name='census_extended'
)


fig = get_column_pair_plot(
    real_data=data,
    synthetic_data=None,
    metadata=metadata,
    column_names=['age', 'occupation'],
    plot_type='heatmap',
)

This errors out.

Expectation

I expect a heatmap to be shown for just the real data (single heatmap square rather than a side-by-side comparison), and the title should be updated to just say Real Data for columns '<name>' and '<name>'

I also expect that this same functionality to work in the opposite case (where I have only synthetic data but not real data).