Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "6589e510fa4e6c442059de2fab84752535de9b23"
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
Show 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): ...@@ -705,4 +705,4 @@ class RoCBertModelIntegrationTest(unittest.TestCase):
# convert to tokens is: ['[CLS]', '巴', '*', '黎', '是', '法', '国', '的', '首', '都', '[SEP]'] # convert to tokens is: ['[CLS]', '巴', '*', '黎', '是', '法', '国', '的', '首', '都', '[SEP]']
expected_output = torch.tensor([[101, 2349, 115, 7944, 3221, 3791, 1744, 4638, 7674, 6963, 102]]) 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