diff --git a/datasets/flwr_datasets/partitioner/distribution_partitioner_test.py b/datasets/flwr_datasets/partitioner/distribution_partitioner_test.py
index 213599f61b47..8003828c4fe5 100644
--- a/datasets/flwr_datasets/partitioner/distribution_partitioner_test.py
+++ b/datasets/flwr_datasets/partitioner/distribution_partitioner_test.py
@@ -97,7 +97,7 @@ def _get_partitioner(
         (20, 1, 200, 10, 5),
     ],
 )
-# pylint: disable=R0913
+# pylint: disable=R0913,R0201
 class TestDistributionPartitioner:
     """Unit tests for DistributionPartitioner."""