chore: improve init file

SCAI-BIO · Sep 27, 2024 · 5b33fab · 5b33fab
1 parent 63ad745
commit 5b33fab
Showing 1 changed file with 40 additions and 3 deletions.
diff --git a/syndat/__init__.py b/syndat/__init__.py
@@ -1,3 +1,40 @@
-from syndat import domain
-from syndat import scores
-from syndat import visualization
+from .visualization import (
+    plot_distributions,
+    plot_correlations,
+    plot_shap_discrimination,
+    plot_categorical_feature,
+    plot_numerical_feature
+)
+
+from .scores import (
+    auc,
+    jsd,
+    discrimination,
+    distribution,
+    correlation,
+)
+
+from .postprocessing import (
+    normalize_scale,
+    normalize_float_precision,
+    assert_minmax,
+) 
+
+__all__ = [
+    # visualization
+    'plot_distributions',
+    'plot_correlations',
+    'plot_shap_discrimination',
+    'plot_categorical_feature',
+    'plot_numerical_feature',
+    # scores
+    'auc',
+    'jsd',
+    'discrimination',
+    'distribution',
+    'correlation',
+    # postprocessing
+    'normalize_scale',
+    'normalize_float_precision',
+    'assert_minmax',
+]