From 902dc7c09cb4fda295ea58136419123ae9a3a1d3 Mon Sep 17 00:00:00 2001
From: chenyu <chenyu@fastmail.com>
Date: Sat, 14 Feb 2026 17:59:00 -0500
Subject: [PATCH] fix test_numpy_parity_and_backward_2d (#14755)

test setup issue, test failed locally with `RUN_SLOW=1`
---
 test/unit/test_indexing.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/unit/test_indexing.py b/test/unit/test_indexing.py
index c95e69193f..31ab4ff124 100644
--- a/test/unit/test_indexing.py
+++ b/test/unit/test_indexing.py
@@ -1000,7 +1000,7 @@ def assert_backward_eq(tensor: Tensor, indexer):
 def get_set_tensor(indexed: Tensor, indexer):
   set_size = indexed[indexer].shape
   set_count = indexed[indexer].numel()
-  set_tensor = Tensor.randint(set_count, high=set_count).reshape(set_size) #.cast(dtypes.float64)
+  set_tensor = Tensor.randint(set_count, high=set_count).reshape(set_size).cast(indexed.dtype)
   return set_tensor
 
 @slow