From 7cc8acdd1f266c95ea76c2b241503a05d6baa991 Mon Sep 17 00:00:00 2001 From: KarhouTam Date: Sun, 22 Sep 2024 00:26:31 +0800 Subject: [PATCH] fix format checks --- .../flwr_datasets/partitioner/image_semantic_partitioner.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/datasets/flwr_datasets/partitioner/image_semantic_partitioner.py b/datasets/flwr_datasets/partitioner/image_semantic_partitioner.py index 41aad1eae77b..7a6d9b7d4990 100644 --- a/datasets/flwr_datasets/partitioner/image_semantic_partitioner.py +++ b/datasets/flwr_datasets/partitioner/image_semantic_partitioner.py @@ -231,6 +231,7 @@ def _determine_partition_id_to_indices_if_needed(self) -> None: (models.efficientnet_b7, models.EfficientNet_B7_Weights.DEFAULT), ] + # pylint: disable=R0913 def _pairwise_kl_div( means_1: torch.Tensor, trils_1: torch.Tensor, @@ -351,7 +352,7 @@ def _subsample(embeddings: NDArrayFloat, num_samples: int) -> NDArrayFloat: gmm.means_, dtype=torch.float, device=device ) label_cluster_trils[current_label] = ( - torch.linalg.cholesky( # type: ignore + torch.linalg.cholesky( # pylint: disable=E1102 torch.from_numpy(gmm.covariances_) ) .float()