fix sig (#1368)

Author: szha

scripts/machine_translation/gnmt.py

@@ -133,7 +133,7 @@ def __call__(self, inputs, states=None, valid_length=None):
         """
         return super(GNMTEncoder, self).__call__(inputs, states, valid_length)
 
-    def forward(self, inputs, states=None, valid_length=None):  #pylint: disable=arguments-differ, missing-docstring
+    def forward(self, inputs, states=None, valid_length=None):  #pylint: missing-docstring
         # TODO(sxjscience) Accelerate the forward using HybridBlock
         _, length, _ = inputs.shape
         new_states = []

src/gluonnlp/model/bert.py

@@ -340,7 +340,7 @@ def __init__(self, *, num_layers=2, units=512, hidden_size=2048,
                     activation=activation, layer_norm_eps=layer_norm_eps)
                 self.transformer_cells.add(cell)
 
-    def __call__(self, inputs, states=None, valid_length=None):  # pylint: disable=arguments-differ
+    def __call__(self, inputs, states=None, valid_length=None):
         """Encode the inputs given the states and valid sequence length.
 
         Parameters

src/gluonnlp/model/seq2seq_encoder_decoder.py

@@ -53,27 +53,27 @@ class Seq2SeqEncoder(Block):
     r"""Base class of the encoders in sequence to sequence learning models.
     """
 
-    def __call__(self, inputs, valid_length=None, states=None):  #pylint: disable=arguments-differ
+    def __call__(self, inputs, states=None, valid_length=None):  #pylint: disable=arguments-differ
         """Encode the input sequence.
 
         Parameters
         ----------
         inputs : NDArray
             The input sequence, Shape (batch_size, length, C_in).
+        states : list of NDArrays or None, default None
+            List that contains the initial states of the encoder.
         valid_length : NDArray or None, default None
             The valid length of the input sequence, Shape (batch_size,). This is used when the
             input sequences are padded. If set to None, all elements in the sequence are used.
-        states : list of NDArrays or None, default None
-            List that contains the initial states of the encoder.
 
         Returns
         -------
         outputs : list
             Outputs of the encoder.
         """
-        return super(Seq2SeqEncoder, self).__call__(inputs, valid_length, states)
+        return super(Seq2SeqEncoder, self).__call__(inputs, states, valid_length)
 
-    def forward(self, inputs, valid_length=None, states=None):  #pylint: disable=arguments-differ
+    def forward(self, inputs, states=None, valid_length=None):  #pylint: disable=arguments-differ
         raise NotImplementedError
 
 

src/gluonnlp/model/transformer.py

@@ -344,7 +344,7 @@ def __init__(self, *, attention_cell='multi_head', num_layers=2, units=512, hidd
                     scaled=scaled, output_attention=output_attention, prefix='transformer%d_' % i)
                 self.transformer_cells.add(cell)
 
-    def __call__(self, inputs, states=None, valid_length=None): #pylint: disable=arguments-differ
+    def __call__(self, inputs, states=None, valid_length=None):
         """Encode the inputs given the states and valid sequence length.
 
         Parameters