From d33c9092e00fe1f9416c346ab1c1a272d90899af Mon Sep 17 00:00:00 2001 From: Adam Narozniak Date: Wed, 13 Mar 2024 18:11:35 +0000 Subject: [PATCH] Replace R0914: Too many local variables --- datasets/flwr_datasets/partitioner/dirichlet_partitioner.py | 4 ++-- datasets/flwr_datasets/partitioner/shard_partitioner.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/datasets/flwr_datasets/partitioner/dirichlet_partitioner.py b/datasets/flwr_datasets/partitioner/dirichlet_partitioner.py index 25d186fdadd1..f3feb2174bde 100644 --- a/datasets/flwr_datasets/partitioner/dirichlet_partitioner.py +++ b/datasets/flwr_datasets/partitioner/dirichlet_partitioner.py @@ -25,7 +25,7 @@ from flwr_datasets.partitioner.partitioner import Partitioner -# pylint: disable=R0902, R0912 +# pylint: disable=R0902, R0912, R0914 class DirichletPartitioner(Partitioner): """Partitioner based on Dirichlet distribution. @@ -196,7 +196,7 @@ def _initialize_alpha( def _determine_partition_id_to_indices_if_needed( self, - ) -> None: # pylint: disable=R0914 + ) -> None: """Create an assignment of indices to the partition indices.""" if self._partition_id_to_indices_determined: return diff --git a/datasets/flwr_datasets/partitioner/shard_partitioner.py b/datasets/flwr_datasets/partitioner/shard_partitioner.py index 22a6abb7f47c..a973f7e5bcb9 100644 --- a/datasets/flwr_datasets/partitioner/shard_partitioner.py +++ b/datasets/flwr_datasets/partitioner/shard_partitioner.py @@ -15,7 +15,7 @@ """Shard partitioner class.""" -# pylint: disable=R0912 +# pylint: disable=R0912, R0914 import math from typing import Dict, List, Optional @@ -198,7 +198,7 @@ def num_partitions(self) -> int: def _determine_partition_id_to_indices_if_needed( self, - ) -> None: # pylint: disable=R0914 + ) -> None: """Assign sample indices to each partition id. This method works on sorted datasets. A "shard" is a part of the dataset of