Internal change

PiperOrigin-RevId: 490370810

official/projects/nhnet/models.py

@@ -311,7 +311,7 @@ class NHNet(Bert2Bert):
 
     return _symbols_to_logits_fn
 
-  def call(self, inputs, mode="training"):
+  def call(self, inputs, mode="training"):  # pytype: disable=signature-mismatch  # overriding-default-value-checks
     input_shape = tf_utils.get_shape_list(inputs["input_ids"], expected_rank=3)
     batch_size, num_docs, len_passage = (input_shape[0], input_shape[1],
                                          input_shape[2])