From 43da68e657f005924b25e6a7caeebcff04434406 Mon Sep 17 00:00:00 2001 From: SangBin Cho Date: Tue, 24 Aug 2021 22:15:34 -0700 Subject: [PATCH] Fix a nightly dask on ray test (#18060) --- release/nightly_tests/dask_on_ray/dask_on_ray_sort.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/release/nightly_tests/dask_on_ray/dask_on_ray_sort.py b/release/nightly_tests/dask_on_ray/dask_on_ray_sort.py index 4fcbdbdf8..8dff48d1f 100644 --- a/release/nightly_tests/dask_on_ray/dask_on_ray_sort.py +++ b/release/nightly_tests/dask_on_ray/dask_on_ray_sort.py @@ -126,8 +126,8 @@ def trial(client, # max_branch=float("inf")).compute( # _ray_enable_progress_bar=True).head(10)) print( - df.set_index("a", shuffle="tasks", - max_branch=float("inf")).compute().head(10)) + df.set_index("a", shuffle="tasks", max_branch=float("inf")).head( + 10, npartitions=-1)) trial_end = time.time() duration = trial_end - trial_start