Drop pyproject.toml entries, start windows fix

Author: adityagoel4512

ndonnx/_funcs.py

@@ -694,7 +694,12 @@ def roll(x, shift, axis=None):
         shift_single = opx.add(opx.const(-sh), len_single)
         # Find the needed element index and then gather from it
         range = opx.cast(
-            opx.range(opx.const(0), len_single, opx.const(1)), to=dtypes.int64
+            opx.range(
+                opx.const(0, dtype=len_single.dtype),
+                len_single,
+                opx.const(1, dtype=len_single.dtype),
+            ),
+            to=dtypes.int64,
         )
         new_indices = opx.mod(opx.add(range, shift_single), len_single)
         x = take(x, _from_corearray(new_indices), axis=ax)

ndonnx/_opset_extensions.py

@@ -640,14 +640,19 @@ def ndindex(shape: _CoreArray, to_reverse=None, axes_permutation=None) -> _CoreA
         axes_indices = [axes_permutation.index(i) for i in builtins.range(rank)]
 
     shape_var = shape.var
+    dtype = shape_var.unwrap_tensor().dtype
     ranges = [
         (
-            op.range(op.const(0), op.gather(shape_var, op.const(i)), op.const(1))
+            op.range(
+                op.const(0, dtype=dtype),
+                op.gather(shape_var, op.const(i)),
+                op.const(1, dtype=dtype),
+            )
             if i not in to_reverse
             else op.range(
-                op.sub(op.gather(shape_var, op.const(i)), op.const(1)),
-                op.const(-1),
-                op.const(-1),
+                op.sub(op.gather(shape_var, op.const(i)), op.const(1, dtype=dtype)),
+                op.const(-1, dtype=dtype),
+                op.const(-1, dtype=dtype),
             )
         )
         for i in builtins.range(rank)

pyproject.toml

@@ -82,12 +82,3 @@ exclude = ["docs/"]
 
 [tool.typos.default]
 extend-ignore-identifiers-re = ["scatter_nd", "arange"]
-
-[tool.pixi.project]
-channels = ["conda-forge"]
-platforms = ["osx-arm64"]
-
-[tool.pixi.pypi-dependencies]
-ndonnx = { path = ".", editable = true }
-
-[tool.pixi.tasks]

tests/ndonnx/test_constant_propagation.py

@@ -67,13 +67,13 @@ def dynamic_masking_model(mode: Literal["lazy", "constant"]):
 
 def constant_indexing_model(mode: Literal["lazy", "constant"]):
     if mode == "constant":
-        a = ndx.asarray([0, 1, 2, 3])
+        a = ndx.asarray([0, 1, 2, 3], dtype=ndx.int64)
     else:
         a = ndx.array(
             shape=("N",),
             dtype=ndx.int64,
         )
-    b = ndx.asarray([5, 7, 8, 8, 9, 9, 234])
+    b = ndx.asarray([5, 7, 8, 8, 9, 9, 234], dtype=ndx.int64)
     idx = ndx.asarray([1, 3, 5, 0])
     result = a * b[idx]
     return ndx.build({"a": a} if mode == "lazy" else {}, {"y": result})

tests/ndonnx/test_core.py

@@ -81,9 +81,11 @@ def test_null_promotion():
 
 
 def test_asarray():
-    a = ndx.asarray([1, 2, 3])
+    a = ndx.asarray([1, 2, 3], dtype=ndx.int64)
     assert a.dtype == ndx.int64
-    np.testing.assert_array_equal(np.array([1, 2, 3]), a.to_numpy(), strict=True)
+    np.testing.assert_array_equal(
+        np.array([1, 2, 3], np.int64), a.to_numpy(), strict=True
+    )
 
 
 def test_asarray_masked():