riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 19:21:31 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

add print statement to avoid code quality problem

Browse Source
This commit is contained in:
Patrick von Platen 2020-03-09 15:31:21 +01:00
parent b12541c4dc
commit efb619235c
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_modeling_tf_transfo_xl.py
View File

@ -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