add __all__

pytorch · Oct 14, 2024 · add2e3f · add2e3f
1 parent 26cf736
commit add2e3f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/ignite/metrics/clustering/calinski_harabasz_score.py b/ignite/metrics/clustering/calinski_harabasz_score.py
@@ -5,6 +5,8 @@
 
 from ignite.metrics.clustering._base import _ClusteringMetricBase
 
+__all__ = ["CalinskiHarabaszScore"]
+
 
 def _calinski_harabasz_score(features: Tensor, labels: Tensor) -> float:
     from sklearn.metrics import calinski_harabasz_score

diff --git a/ignite/metrics/clustering/davies_bouldin_score.py b/ignite/metrics/clustering/davies_bouldin_score.py
@@ -5,6 +5,8 @@
 
 from ignite.metrics.clustering._base import _ClusteringMetricBase
 
+__all__ = ["DaviesBouldinScore"]
+
 
 def _davies_bouldin_score(features: Tensor, labels: Tensor) -> float:
     from sklearn.metrics import davies_bouldin_score

diff --git a/ignite/metrics/clustering/silhouette_score.py b/ignite/metrics/clustering/silhouette_score.py
@@ -5,6 +5,8 @@
 
 from ignite.metrics.clustering._base import _ClusteringMetricBase
 
+__all__ = ["SilhouetteScore"]
+
 
 def _get_silhouette_score(**kwargs: Any) -> Callable[[Tensor, Tensor], float]:
     from sklearn.metrics import silhouette_score