diff --git a/sdmetrics/__init__.py b/sdmetrics/__init__.py index 8bc74625..6bcfd379 100644 --- a/sdmetrics/__init__.py +++ b/sdmetrics/__init__.py @@ -15,7 +15,7 @@ from pkg_resources import iter_entry_points from sdmetrics import ( - column_pairs, demos, goal, multi_table, single_column, single_table, timeseries) + column_pairs, demos, goal, multi_table, reports, single_column, single_table, timeseries) from sdmetrics.demos import load_demo __all__ = [ @@ -27,6 +27,7 @@ 'single_column', 'single_table', 'timeseries', + 'reports', ] diff --git a/sdmetrics/reports/__init__.py b/sdmetrics/reports/__init__.py index 26bffc22..df0ff46b 100644 --- a/sdmetrics/reports/__init__.py +++ b/sdmetrics/reports/__init__.py @@ -1,6 +1,14 @@ """Reports for sdmetrics.""" +from sdmetrics.reports.multi_table import DiagnosticReport as MultiTableDiagnosticReport +from sdmetrics.reports.multi_table import QualityReport as MultiTableQualityReport +from sdmetrics.reports.single_table import DiagnosticReport as SingleTableDiagnosticReport +from sdmetrics.reports.single_table import QualityReport as SingleTableQualityReport from sdmetrics.reports.utils import get_column_pair_plot __all__ = [ 'get_column_pair_plot', + 'SingleTableQualityReport', + 'SingleTableDiagnosticReport', + 'MultiTableQualityReport', + 'MultiTableDiagnosticReport', ]