From 0bdd4730af8c215545821049ea9f794519c13fa8 Mon Sep 17 00:00:00 2001 From: Mohammad Miadh Angkad <176301910+mmangkad@users.noreply.github.com> Date: Thu, 23 Jul 2026 00:11:25 +0800 Subject: [PATCH] [CI] Fix failures on main (#32091) --- .../model_runner_components/test_ngram_embedding_manager.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/registered/unit/model_executor/model_runner_components/test_ngram_embedding_manager.py b/test/registered/unit/model_executor/model_runner_components/test_ngram_embedding_manager.py index 3010d86db..57270eea3 100644 --- a/test/registered/unit/model_executor/model_runner_components/test_ngram_embedding_manager.py +++ b/test/registered/unit/model_executor/model_runner_components/test_ngram_embedding_manager.py @@ -110,9 +110,9 @@ class TestNgramTokenTableUpdate(CustomTestCase): kwargs["tokens"], torch.tensor([101, 202, 303], dtype=torch.int32) ) ) - self.assertIs(kwargs["row_indices"], req_pool_indices) - self.assertIs(kwargs["column_starts"], info.out_column_starts) - self.assertIs(kwargs["req_lens"], info.out_req_lens) + self.assertTrue(torch.equal(kwargs["row_indices"], req_pool_indices)) + self.assertTrue(torch.equal(kwargs["column_starts"], info.out_column_starts)) + self.assertTrue(torch.equal(kwargs["req_lens"], info.out_req_lens)) self.assertTrue( torch.equal( info.out_column_starts, torch.tensor([11, 22, 33], dtype=torch.int32)