[dask-on-ray] Fix Dask-on-Ray test: Python 3 dictionary .values() is a view, and is not indexable (#13945)

This commit is contained in:
Clark Zinzow
2021-02-05 22:21:41 -07:00
committed by GitHub
parent ea4154df80
commit f070b3c9a9
+3 -1
View File
@@ -35,7 +35,9 @@ def test_ray_dask_basic(ray_start_regular_shared):
def test_ray_dask_persist(ray_start_regular_shared):
arr = da.ones(5) + 2
result = arr.persist(scheduler=ray_dask_get)
np.testing.assert_array_equal(result.dask.values()[0], np.ones(5) + 2)
np.testing.assert_array_equal(
next(iter(result.dask.values())),
np.ones(5) + 2)
if __name__ == "__main__":