diff --git a/datasets/flwr_datasets/partitioner/vertical_partitioner_utils.py b/datasets/flwr_datasets/partitioner/vertical_partitioner_utils.py
index 097cde81e576..d232f0c80207 100644
--- a/datasets/flwr_datasets/partitioner/vertical_partitioner_utils.py
+++ b/datasets/flwr_datasets/partitioner/vertical_partitioner_utils.py
@@ -47,7 +47,7 @@ def _list_split(lst: list[Any], num_sublists: int) -> list[list[Any]]:
     return sublists
 
 
-def _add_active_party_columns(
+def _add_active_party_columns(  # pylint: disable=R0912
     active_party_columns: Union[str, list[str]],
     active_party_columns_mode: Union[
         Literal[