From 3fdae3fe1a45dfc337d1b7cae7676bdd1eba9809 Mon Sep 17 00:00:00 2001
From: Adam Narozniak <adam@flower.dev>
Date: Thu, 19 Dec 2024 14:30:32 +0100
Subject: [PATCH] Fix tests

---
 .../flwr_datasets/partitioner/vertical_partitioner_utils.py     | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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[