diff --git a/datasets/flwr_datasets/partitioner/vertical_even_partitioner_test.py b/datasets/flwr_datasets/partitioner/vertical_even_partitioner_test.py
index 0eae437eb0f8..aa93db7a9fd5 100644
--- a/datasets/flwr_datasets/partitioner/vertical_even_partitioner_test.py
+++ b/datasets/flwr_datasets/partitioner/vertical_even_partitioner_test.py
@@ -50,36 +50,21 @@ def test_init_with_invalid_active_party_mode(self) -> None:
 
     def test_init_with_non_string_drop_columns(self) -> None:
         """Test initialization with non-string elements in drop_columns."""
-        with self.assertRaises(ValueError):
+        with self.assertRaises(TypeError):
             VerticalEvenPartitioner(num_partitions=2, drop_columns=[1, "a", 3])
 
-    def test_init_with_non_list_drop_columns(self) -> None:
-        """Test initialization with non-list elements in drop_columns."""
-        with self.assertRaises(ValueError):
-            VerticalEvenPartitioner(num_partitions=2, drop_columns="a")
-
     def test_init_with_non_string_shared_columns(self) -> None:
         """Test initialization with non-string elements in shared_columns."""
-        with self.assertRaises(ValueError):
+        with self.assertRaises(TypeError):
             VerticalEvenPartitioner(num_partitions=2, shared_columns=["col1", 123])
 
-    def test_init_with_non_list_shared_columns(self) -> None:
-        """Test initialization with non-list elements in shared_columns."""
-        with self.assertRaises(ValueError):
-            VerticalEvenPartitioner(num_partitions=2, shared_columns="col1")
-
     def test_init_with_non_string_active_party_column(self) -> None:
         """Test initialization with non-string elements in active_party_column."""
-        with self.assertRaises(ValueError):
+        with self.assertRaises(TypeError):
             VerticalEvenPartitioner(
                 num_partitions=2, active_party_columns=["col1", None]
             )
 
-    def test_init_with_non_list_active_party_column(self) -> None:
-        """Test initialization with non-list elements in active_party_column."""
-        with self.assertRaises(ValueError):
-            VerticalEvenPartitioner(num_partitions=2, active_party_columns="col1")
-
     def test_partitioning_basic(self) -> None:
         """Test basic partitioning with no special columns or dropping."""
         columns = ["feature1", "feature2", "feature3", "feature4"]