[ROC_BERT] Make CI happy (#20175)

* fix slow test * Update tests/models/roc_bert/test_modeling_roc_bert.py

[ROC_BERT] Make CI happy (#20175)
* fix slow test * Update tests/models/roc_bert/test_modeling_roc_bert.py
8dcf494e · Younes Belkada · GitHub · 7b55bb45 · 8dcf494e
Unverified Commit 8dcf494e authored Nov 14, 2022 by Younes Belkada Committed by GitHub Nov 14, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/models/roc_bert/test_modeling_roc_bert.py tests/models/roc_bert/test_modeling_roc_bert.py +1 -1

No files found.
--- a/tests/models/roc_bert/test_modeling_roc_bert.py
+++ b/tests/models/roc_bert/test_modeling_roc_bert.py
@@ -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)