fix format checks

adap · Sep 21, 2024 · 7cc8acd · 7cc8acd
1 parent 1135913
commit 7cc8acd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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()