From 34b61dfd605910a54de9293e1894302ef8ccbfa7 Mon Sep 17 00:00:00 2001
From: kralka <karelkral@google.com>
Date: Mon, 7 Oct 2024 10:34:31 +0200
Subject: [PATCH] Increase the maximum number of arguments (#27)

Set pylint to be more lenient.
---
 .pylintrc                           | 3 +++
 src/sedpack/io/dataset_iteration.py | 8 ++++----
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/.pylintrc b/.pylintrc
index a9015a7f..8e5954a7 100644
--- a/.pylintrc
+++ b/.pylintrc
@@ -440,3 +440,6 @@ valid-metaclass-classmethod-first-arg=mcs
 overgeneral-exceptions=builtins.StandardError,
                        builtins.Exception,
                        builtins.BaseException
+
+# Set number of arguments for too-many-arguments
+max-args = 10
diff --git a/src/sedpack/io/dataset_iteration.py b/src/sedpack/io/dataset_iteration.py
index e79b5fa8..648055ea 100644
--- a/src/sedpack/io/dataset_iteration.py
+++ b/src/sedpack/io/dataset_iteration.py
@@ -261,7 +261,7 @@ def as_tfdataset(  # pylint: disable=too-many-arguments
 
         return tf_dataset
 
-    async def as_numpy_iterator_async(  # pylint: disable=too-many-arguments
+    async def as_numpy_iterator_async(
         self,
         *,
         split: SplitT,
@@ -354,7 +354,7 @@ async def as_numpy_iterator_async(  # pylint: disable=too-many-arguments
         async for example in example_iterator:
             yield example
 
-    def _as_numpy_common(  # pylint: disable=too-many-arguments
+    def _as_numpy_common(
         self,
         *,
         split: SplitT,
@@ -405,7 +405,7 @@ def _as_numpy_common(  # pylint: disable=too-many-arguments
                 buffer_size=len(shard_paths))
         return shard_paths_iterator
 
-    def as_numpy_iterator_concurrent(  # pylint: disable=too-many-arguments
+    def as_numpy_iterator_concurrent(
         self,
         *,
         split: SplitT,
@@ -508,7 +508,7 @@ def as_numpy_iterator_concurrent(  # pylint: disable=too-many-arguments
                             itertools.islice(shard_paths_iterator,
                                              file_parallelism))
 
-    def as_numpy_iterator(  # pylint: disable=too-many-arguments
+    def as_numpy_iterator(
         self,
         *,
         split: SplitT,