Removing useless args and kwargs arguments in tasks.

tasks/cs/bucket_sort.py

@@ -38,15 +38,12 @@ class BucketSort(task.GeneralizationTask):
     1110001   ->  0001111   (with `vocab_size = 2`)
   """
 
-  def __init__(self, *args, vocab_size: int = 5, **kwargs) -> None:
+  def __init__(self, vocab_size: int = 5) -> None:
     """Initializes the task.
 
     Args:
-      *args: The args for the base task class.
       vocab_size: The size of the alphabet. We use 5 in the paper.
-      **kwargs: The kwargs for the base task class.
     """
-    super().__init__(*args, **kwargs)
     self._vocab_size = vocab_size
 
   @functools.partial(jax.jit, static_argnums=(0, 2, 3))

tasks/cs/duplicate_string.py

@@ -40,16 +40,12 @@ class DuplicateString(task.GeneralizationTask):
   Note that the sampling is jittable so this task is fast.
   """
 
-  def __init__(self, *args, vocab_size: int = 2, **kwargs):
+  def __init__(self, vocab_size: int = 2) -> None:
     """Initializes the remember_string task.
 
     Args:
       vocab_size: The size of the alphabet. We use 2 in the paper.
-      *args: Args for the base task class.
-      **kwargs: Kwargs for the base task class.
     """
-    super().__init__(*args, **kwargs)
-
     self._vocab_size = vocab_size
 
   @functools.partial(jax.jit, static_argnums=(0, 2, 3))

tasks/cs/odds_first.py

@@ -40,16 +40,12 @@ class OddsFirst(task.GeneralizationTask):
   Note that the sampling is jittable so this task is fast.
   """
 
-  def __init__(self, *args, vocab_size: int = 2, **kwargs):
+  def __init__(self, vocab_size: int = 2) -> None:
     """Initializes the odds_first task.
 
     Args:
       vocab_size: The size of the alphabet. We use 2 in the paper.
-      *args: Args for the base task class.
-      **kwargs: Kwargs for the base task class.
     """
-    super().__init__(*args, **kwargs)
-
     self._vocab_size = vocab_size
 
   @functools.partial(jax.jit, static_argnums=(0, 2, 3))

tasks/dcf/modular_arithmetic_brackets.py

@@ -143,16 +143,13 @@ def generate_raw_dataset(
 class ModularArithmeticBrackets(task.GeneralizationTask):
   """A task with the goal of reducing an arithmetic expression with brackets."""
 
-  def __init__(self, *args, modulus: int = 5, mult: bool = False, **kwargs):
+  def __init__(self, modulus: int = 5, mult: bool = False) -> None:
     """Initializes the modular arithmetic task.
 
     Args:
-      *args: Args for the base task class.
       modulus: The modulus used for the computation. We use 5 in the paper.
       mult: Whether to add multiplication or use only '+' and '-'.
-      **kwargs: Kwargs for the base task class.
     """
-    super().__init__(*args, **kwargs)
     self._modulus = modulus
     self._mult = mult
 

tasks/dcf/solve_equation.py

@@ -125,15 +125,12 @@ class SolveEquation(task.GeneralizationTask):
   multiple solutions (multiplication by zero).
   """
 
-  def __init__(self, *args, modulus: int = 5, **kwargs):
+  def __init__(self, modulus: int = 5) -> None:
     """Initializes the modular arithmetic task.
 
     Args:
-      *args: Args for the base task class.
       modulus: The modulus used for the computation. We use 5 in the paper.
-      **kwargs: Kwargs for the base task class.
     """
-    super().__init__(*args, **kwargs)
     self._modulus = modulus
 
   def sample_batch(

tasks/regular/modular_arithmetic.py

@@ -132,22 +132,18 @@ class ModularArithmetic(task.GeneralizationTask):
   Note that the input strings are always of odd length.
   """
 
-  def __init__(self,
-               *args,
-               modulus: int = 5,
-               operators: Optional[Sequence[str]] = None,
-               **kwargs):
+  def __init__(
+    self,
+    modulus: int = 5,
+     operators: Optional[Sequence[str]] = None,
+  ) -> None:
     """Initializes the modular arithmetic task.
 
     Args:
-      *args: Args for the base task class.
       modulus: The modulus used for the computation. We use 5 in the paper.
       operators: Operators to be used in the sequences. By default it's None,
         meaning all operators available are used.
-      **kwargs: Kwargs for the base task class.
     """
-    super().__init__(*args, **kwargs)
-
     self._modulus = modulus
     if operators is None:
       operators = ('+', '*', '-')