From 3c8daa9a75fccdbfdb4ef3d523e61dc616c8bfab Mon Sep 17 00:00:00 2001 From: chenyu Date: Wed, 22 Apr 2026 14:56:37 -0400 Subject: [PATCH] update test_where_removal (#15872) don't use UOp.ufix for const_like, it will broadcast dtype soon --- test/null/test_uop_symbolic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/null/test_uop_symbolic.py b/test/null/test_uop_symbolic.py index 08c2ff6c69..51c4ebe714 100644 --- a/test/null/test_uop_symbolic.py +++ b/test/null/test_uop_symbolic.py @@ -853,7 +853,7 @@ class TestSymbolic(unittest.TestCase): def test_where_removal(self): cond = Variable("a", 0, 3) < 2 - u1, u0 = cond.ufix(1), cond.ufix(0) + u1, u0 = cond.const_like(True), cond.const_like(False) self.helper_test_variable(cond, 0, 1, "(a<2)") self.helper_test_variable(cond.where(u1, u0), 0, 1, "(a<2)") self.helper_test_variable(cond.where(u1, u0).where(u1, u0), 0, 1, "(a<2)")