diff --git a/sdmetrics/reports/single_table/_properties/__init__.py b/sdmetrics/reports/single_table/_properties/__init__.py index ef06c1aa..c4a7831a 100644 --- a/sdmetrics/reports/single_table/_properties/__init__.py +++ b/sdmetrics/reports/single_table/_properties/__init__.py @@ -1,7 +1,9 @@ """Single table properties for sdmetrics.""" from sdmetrics.reports.single_table._properties.base import BaseSingleTableProperty +from sdmetrics.reports.single_table._properties.column_shapes import ColumnShapes __all__ = [ - 'BaseSingleTableProperty' + 'BaseSingleTableProperty', + 'ColumnShapes', ] diff --git a/sdmetrics/reports/single_table/_properties/column_shapes.py b/sdmetrics/reports/single_table/_properties/column_shapes.py index 6467a265..a08e353f 100644 --- a/sdmetrics/reports/single_table/_properties/column_shapes.py +++ b/sdmetrics/reports/single_table/_properties/column_shapes.py @@ -5,8 +5,7 @@ import plotly.express as px from sdmetrics.reports.single_table._properties import BaseSingleTableProperty -from sdmetrics.single_column.statistical.kscomplement import KSComplement -from sdmetrics.single_column.statistical.tv_complement import TVComplement +from sdmetrics.single_column import KSComplement, TVComplement class ColumnShapes(BaseSingleTableProperty):