imports

sdv-dev · Jun 21, 2023 · 42b545a · 42b545a
1 parent 6e5c569
commit 42b545a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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):