fixed all tests, still need to check ctrl tf and pt and xlm tf

2020-03-08 21:45:55 +01:00
parent b4a3a64744
commit fbd02d4693
7 changed files with 51 additions and 49 deletions
--- a/tests/test_modeling_tf_xlnet.py
+++ b/tests/test_modeling_tf_xlnet.py
@@ -814,4 +814,4 @@ class TFXLNetModelLanguageGenerationTest(unittest.TestCase):

        output_ids = model.generate(input_ids, max_length=200, do_sample=False)

-        self.assertListEqual(output_ids[0].tolist(), expected_output_ids)
+        self.assertListEqual(output_ids[0].numpy().tolist(), expected_output_ids)