Skip to content

GitLab

Unverified Commit 8dcf494e authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

[ROC_BERT] Make CI happy (#20175) 

* fix slow test

* Update tests/models/roc_bert/test_modeling_roc_bert.py
parent 7b55bb45
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/roc_bert/test_modeling_roc_bert.py
View file @ 8dcf494e
......@@ -705,4 +705,4 @@ class RoCBertModelIntegrationTest(unittest.TestCase):
# convert to tokens is: ['[CLS]', '巴', '*', '黎', '是', '法', '国', '的', '首', '都', '[SEP]']
expected_output = torch.tensor([[101, 2349, 115, 7944, 3221, 3791, 1744, 4638, 7674, 6963, 102]])
self.assertTrue(output_ids, expected_output)
self.assertEqual(output_ids, expected_output)
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