diff --git a/src/torchmetrics/functional/clustering/adjusted_rand_score.py b/src/torchmetrics/functional/clustering/adjusted_rand_score.py index 38412edb437..5733d8b49d7 100644 --- a/src/torchmetrics/functional/clustering/adjusted_rand_score.py +++ b/src/torchmetrics/functional/clustering/adjusted_rand_score.py @@ -15,8 +15,8 @@ from torch import Tensor from torchmetrics.functional.clustering.utils import ( - calculate_pair_cluster_confusion_matrix, calculate_contingency_matrix, + calculate_pair_cluster_confusion_matrix, check_cluster_labels, ) diff --git a/src/torchmetrics/functional/clustering/rand_score.py b/src/torchmetrics/functional/clustering/rand_score.py index e53d4e67cc5..51b719641fd 100644 --- a/src/torchmetrics/functional/clustering/rand_score.py +++ b/src/torchmetrics/functional/clustering/rand_score.py @@ -15,8 +15,8 @@ from torch import Tensor from torchmetrics.functional.clustering.utils import ( - calculate_pair_cluster_confusion_matrix, calculate_contingency_matrix, + calculate_pair_cluster_confusion_matrix, check_cluster_labels, ) diff --git a/tests/unittests/clustering/test_utils.py b/tests/unittests/clustering/test_utils.py index 9c07fd88ac9..6fc8577f15c 100644 --- a/tests/unittests/clustering/test_utils.py +++ b/tests/unittests/clustering/test_utils.py @@ -21,10 +21,10 @@ from sklearn.metrics.cluster import pair_confusion_matrix as sklearn_pair_confusion_matrix from sklearn.metrics.cluster._supervised import _generalized_average as sklearn_generalized_average from torchmetrics.functional.clustering.utils import ( - calculate_pair_cluster_confusion_matrix, calculate_contingency_matrix, calculate_entropy, calculate_generalized_mean, + calculate_pair_cluster_confusion_matrix, ) from unittests import BATCH_SIZE, NUM_BATCHES