Clean special token init in modeling_....py (#3264)

* make style * fix conflicts

Clean special token init in modeling_....py (#3264)
* make style * fix conflicts
95e00d08 · Patrick von Platen · GitHub · 8becb732 · 95e00d08 · 95e00d08
Unverified Commit 95e00d08 authored Mar 20, 2020 by Patrick von Platen Committed by GitHub Mar 20, 2020
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tests/test_modeling_tf_transfo_xl.py tests/test_modeling_tf_transfo_xl.py +1 -1

tests/test_modeling_transfo_xl.py tests/test_modeling_transfo_xl.py +1 -1

No files found.
--- a/tests/test_modeling_tf_transfo_xl.py
+++ b/tests/test_modeling_tf_transfo_xl.py
@@ -107,7 +107,7 @@ class TFTransfoXLModelTest(TFModelTesterMixin, unittest.TestCase):
                d_inner=self.d_inner,
                div_val=self.div_val,
                n_layer=self.num_hidden_layers,
-                eos_token_ids=self.eos_token_id,
+                eos_token_id=self.eos_token_id,
            )

            return (config, input_ids_1, input_ids_2, lm_labels)

--- a/tests/test_modeling_transfo_xl.py
+++ b/tests/test_modeling_transfo_xl.py
@@ -103,7 +103,7 @@ class TransfoXLModelTest(ModelTesterMixin, unittest.TestCase):
                d_inner=self.d_inner,
                div_val=self.div_val,
                n_layer=self.num_hidden_layers,
-                eos_token_ids=self.eos_token_id,
+                eos_token_id=self.eos_token_id,
            )

            return (config, input_ids_1, input_ids_2, lm_labels)