Update test_indexing.py

npolina4 · web-flow · commit 6e2c3c72b112 · 2024-10-25T00:45:46.000-07:00
diff --git a/tests/test_indexing.py b/tests/test_indexing.py
@@ -728,6 +728,8 @@ class TestTakeAlongAxis:
         ],
     )
     def test_argequivalent(self, func, argfunc, kwargs):
+        # TODO: to roll back the change once the issue with CUDA support is resolved for random
+        # a = dpnp.random.random(size=(3, 4, 5))
         a = dpnp.asarray(numpy.random.random(size=(3, 4, 5)))
 
         for axis in list(range(a.ndim)) + [None]:

Original file line number	Diff line number	Diff line change
`@@ -728,6 +728,8 @@ class TestTakeAlongAxis:`
`728`	`728`	`],`
`729`	`729`	`)`
`730`	`730`	`def test_argequivalent(self, func, argfunc, kwargs):`
	`731`	`+ # TODO: to roll back the change once the issue with CUDA support is resolved for random`
	`732`	`+ # a = dpnp.random.random(size=(3, 4, 5))`
`731`	`733`	`a = dpnp.asarray(numpy.random.random(size=(3, 4, 5)))`
`732`	`734`
`733`	`735`	`for axis in list(range(a.ndim)) + [None]:`