-S

Use cupy.flatiter in test
rapidsai · Jun 27, 2024 · d750fa6 · d750fa6
1 parent f3d4e31
commit d750fa6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cudf/cudf_pandas_tests/test_cudf_pandas.py b/python/cudf/cudf_pandas_tests/test_cudf_pandas.py
@@ -1541,5 +1541,5 @@ def test_numpy_cupy_flatiter(series):
     _, s = series
     arr = s.values
 
-    assert type(arr.flat._fsproxy_fast) == cp._indexing.iterate.flatiter
+    assert type(arr.flat._fsproxy_fast) == cp.flatiter
     assert type(arr.flat._fsproxy_slow) == np.flatiter