pylint

official/nlp/modeling/models/seq2seq_transformer.py

@@ -53,22 +53,22 @@ def create_model(params, is_train):
   decoder_layer = TransformerDecoder(**encdec_kwargs)
 
   model_kwargs = dict(
-          vocab_size=params["vocab_size"],
-          hidden_size=params["hidden_size"],
-          dropout_rate=params["layer_postprocess_dropout"],
-          padded_decode=params["padded_decode"],
-          num_replicas=params["num_replicas"],
-          decode_batch_size=params["decode_batch_size"],
-          decode_max_length=params["decode_max_length"],
-          dtype=params["dtype"],
-          extra_decode_length=params["extra_decode_length"],
-          num_heads=params["num_heads"],
-          num_layers=params["num_hidden_layers"],
-          beam_size=params["beam_size"],
-          alpha=params["alpha"],
-          encoder_layer=encoder_layer,
-          decoder_layer=decoder_layer,
-          name="transformer_v2")
+      vocab_size=params["vocab_size"],
+      hidden_size=params["hidden_size"],
+      dropout_rate=params["layer_postprocess_dropout"],
+      padded_decode=params["padded_decode"],
+      num_replicas=params["num_replicas"],
+      decode_batch_size=params["decode_batch_size"],
+      decode_max_length=params["decode_max_length"],
+      dtype=params["dtype"],
+      extra_decode_length=params["extra_decode_length"],
+      num_heads=params["num_heads"],
+      num_layers=params["num_hidden_layers"],
+      beam_size=params["beam_size"],
+      alpha=params["alpha"],
+      encoder_layer=encoder_layer,
+      decoder_layer=decoder_layer,
+      name="transformer_v2")
 
   with tf.name_scope("model"):
     if is_train:

official/nlp/modeling/models/seq2seq_transformer_test.py

-# Copyright 2018 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2020 The TensorFlow Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -14,10 +14,6 @@
 # ==============================================================================
 """Test Transformer model."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import tensorflow as tf
 
 from official.nlp.modeling.models import seq2seq_transformer