Fix a nightly dask on ray test (#18060)

This commit is contained in:
SangBin Cho 2021-08-24 22:15:34 -07:00 committed by GitHub
parent 08193b8e0a
commit 43da68e657
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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