Fix formatting errors

adap · Dec 13, 2024 · 7b56c1f · 7b56c1f
1 parent 6ac5a7b
commit 7b56c1f
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/datasets/flwr_datasets/partitioner/vertical_even_partitioner.py b/datasets/flwr_datasets/partitioner/vertical_even_partitioner.py
@@ -14,6 +14,7 @@
 # ==============================================================================
 """VerticalEvenPartitioner class."""
 # flake8: noqa: E501
+# pylint: disable=C0301, R0902, R0913
 from typing import Literal, Optional, Union
 
 import numpy as np

diff --git a/datasets/flwr_datasets/partitioner/vertical_even_partitioner_test.py b/datasets/flwr_datasets/partitioner/vertical_even_partitioner_test.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 # ==============================================================================
 """VerticalEvenPartitioner class tests."""
-# mypy: disable-error-code=list-item,arg-type
+# mypy: disable-error-code=list-item
 import unittest
 
 import numpy as np
@@ -44,7 +44,8 @@ def test_init_with_invalid_active_party_mode(self) -> None:
         """Test initialization with invalid active_party_columns_mode."""
         with self.assertRaises(ValueError):
             VerticalEvenPartitioner(
-                num_partitions=2, active_party_columns_mode="invalid_mode"
+                num_partitions=2,
+                active_party_columns_mode="invalid_mode",  # type: ignore[arg-type]
             )
 
     def test_init_with_non_string_drop_columns(self) -> None:

diff --git a/datasets/flwr_datasets/partitioner/vertical_partitioner_utils.py b/datasets/flwr_datasets/partitioner/vertical_partitioner_utils.py
@@ -14,6 +14,7 @@
 # ==============================================================================
 """VerticalPartitioner utils.py."""
 # flake8: noqa: E501
+# pylint: disable=C0301
 from typing import Any, Literal, Union