From b73dd1a0e441e2fc215c013dc4ea2fed65db6c5f Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 9 Mar 2020 11:34:31 +0100 Subject: [PATCH] fix typo in test xlm tf --- tests/test_modeling_tf_xlm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_modeling_tf_xlm.py b/tests/test_modeling_tf_xlm.py index d98fc44131..5048930b56 100644 --- a/tests/test_modeling_tf_xlm.py +++ b/tests/test_modeling_tf_xlm.py @@ -341,5 +341,5 @@ class TFXLMModelLanguageGenerationTest(unittest.TestCase): 447, ] # the president the president the president the president the president the president the president the president the president the president # TODO(PVP): this and other input_ids I tried for generation give pretty bad results. Not sure why. Model might just not be made for auto-regressive inference - output_ids = model.generate(input_ids) - self.assertListEqual(output_ids[0].numpy().tolist(), expected_output_ids, do_sample=False) + output_ids = model.generate(input_ids, do_sample=False) + self.assertListEqual(output_ids[0].numpy().tolist(), expected_output_ids)