SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[ROC_BERT] Make CI happy (#20175)

Browse Source 
* fix slow test

* Update tests/models/roc_bert/test_modeling_roc_bert.py
This commit is contained in:
Younes Belkada
2022-11-14 18:04:25 +01:00
committed by GitHub
parent 7b55bb4540
commit 8dcf494ef1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/roc_bert/test_modeling_roc_bert.py
View File

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