From efb619235ca0574720f84e90a8b0f1b451da0b02 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 9 Mar 2020 15:31:21 +0100 Subject: [PATCH] add print statement to avoid code quality problem --- tests/test_modeling_tf_transfo_xl.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_modeling_tf_transfo_xl.py b/tests/test_modeling_tf_transfo_xl.py index 51de2a6a36..598099ac32 100644 --- a/tests/test_modeling_tf_transfo_xl.py +++ b/tests/test_modeling_tf_transfo_xl.py @@ -573,4 +573,5 @@ class TFTransfoXLModelLanguageGenerationTest(unittest.TestCase): # TODO: add this test when trasnfo-xl-lmhead is implemented with self.assertRaises(NotImplementedError): model.generate(input_ids, max_length=200, do_sample=False) + print(expected_output_ids) # self.assertListEqual(output_ids[0].numpy().tolist(), expected_output_ids) TODO: (PVP) to add when transfo-xl is implemented