From 8dcf494ef1c481000e5b48eef331f1c812170f62 Mon Sep 17 00:00:00 2001
From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>
Date: Mon, 14 Nov 2022 18:04:25 +0100
Subject: [PATCH] [ROC_BERT] Make CI happy (#20175)

* fix slow test

* Update tests/models/roc_bert/test_modeling_roc_bert.py
---
 tests/models/roc_bert/test_modeling_roc_bert.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/models/roc_bert/test_modeling_roc_bert.py b/tests/models/roc_bert/test_modeling_roc_bert.py
index 2a08b891c9..cf3218ac83 100644
--- 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)