From f66aff799e10e9909c3b9ddc7e2692e1f33572c7 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 26 Sep 2023 07:59:14 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- src/torchmetrics/functional/clustering/adjusted_rand_score.py | 2 +- src/torchmetrics/functional/clustering/rand_score.py | 2 +- tests/unittests/clustering/test_utils.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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