Fix CI - rewrite rule test of scatternd (#1936)

Fix ci
microsoft · Nov 7, 2024 · edfa265 · edfa265
1 parent 6ee7c21
commit edfa265
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxscript/rewriter/collapse_slices_test.py b/onnxscript/rewriter/collapse_slices_test.py
@@ -76,7 +76,7 @@ def test_scatternd_is_redundant_when_it_is_updating_the_whole_input_in_order(sel
         """
         )
         # Use inserted initializers to avoid manually coding the large constants
-        indices = np.arange(112).reshape(112, 1)
+        indices = np.arange(112).reshape(112, 1).astype(np.int64)
         model = ir.serde.deserialize_model(model_proto)
         # from numpy to ir.Tensor
         indices_ir_tensor = ir.Tensor(