From 9495737d828ce9ec353fa61e2b8e1ba547dc8bc4 Mon Sep 17 00:00:00 2001 From: Brayden Zhong Date: Thu, 25 Jun 2026 10:35:34 -0700 Subject: [PATCH] Fix CI broken by #28450 (#29308) Co-authored-by: Brayden Zhong --- test/registered/moe/test_topk_padded_region.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/registered/moe/test_topk_padded_region.py b/test/registered/moe/test_topk_padded_region.py index 03d258a3a..616802cf6 100644 --- a/test/registered/moe/test_topk_padded_region.py +++ b/test/registered/moe/test_topk_padded_region.py @@ -109,15 +109,15 @@ class TestTopkPaddedRegion(CustomTestCase): def test_invalid_pad_count_tensor_raises(self): x = torch.rand((8, 8), device=self.DEVICE, dtype=torch.float32) - with self.assertRaises(AssertionError): + with self.assertRaises(TypeError): _fill_padded_rows(x, 4, 0.0) # python int, not a tensor - with self.assertRaises(AssertionError): + with self.assertRaises(ValueError): _fill_padded_rows( x, torch.tensor([1, 2], device=self.DEVICE, dtype=torch.int32), 0.0, ) - with self.assertRaises(AssertionError): + with self.assertRaises(TypeError): _fill_padded_rows( x, torch.tensor(4.0, device=self.DEVICE, dtype=torch.float32),