Skip to content

GitLab

Commit efb61923 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

add print statement to avoid code quality problem

parent b12541c4
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/test_modeling_tf_transfo_xl.py
View file @ efb61923
...@@ -573,4 +573,5 @@ class TFTransfoXLModelLanguageGenerationTest(unittest.TestCase): ...@@ -573,4 +573,5 @@ class TFTransfoXLModelLanguageGenerationTest(unittest.TestCase):
# TODO: add this test when trasnfo-xl-lmhead is implemented # TODO: add this test when trasnfo-xl-lmhead is implemented
with self.assertRaises(NotImplementedError): with self.assertRaises(NotImplementedError):
model.generate(input_ids, max_length=200, do_sample=False) 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 # self.assertListEqual(output_ids[0].numpy().tolist(), expected_output_ids) TODO: (PVP) to add when transfo-xl is implemented
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment