expose reports module and individual reports

sdv-dev · Oct 2, 2023 · e9de32f · e9de32f
1 parent 35978c4
commit e9de32f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git 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
@@ -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',
 ]