put(): a: ArrayLike -> a: NDArray

Prevents normalising non-ndarray arguments

Author: honno

torch_np/_funcs_impl.py

@@ -910,7 +910,7 @@ def take_along_axis(arr: ArrayLike, indices: ArrayLike, axis):
 
 
 def put(
-    a: ArrayLike,
+    a: NDArray,
     ind: ArrayLike,
     v: ArrayLike,
     mode: NotImplementedType = "raise",

torch_np/tests/numpy_tests/core/test_multiarray.py

@@ -2637,7 +2637,6 @@ def test_trace(self):
         ret = a.trace(out=out)
         assert ret is out
 
-    @pytest.mark.xfail(reason="TODO: implement put")
     def test_put(self):
         icodes = np.typecodes['AllInteger']
         fcodes = np.typecodes['AllFloat']
@@ -2670,7 +2669,6 @@ def test_put(self):
         # when calling np.put, make sure a
         # TypeError is raised if the object
         # isn't an ndarray
-        pytest.xfail(reason="XXX: Argument normalisation prevents catching this")
         bad_array = [1, 2, 3]
         assert_raises(TypeError, np.put, bad_array, [0, 2], 5)