|
40 | 40 | import dask.bag as db |
41 | 41 | from dask import delayed |
42 | 42 | from dask.optimization import SubgraphCallable |
43 | | -from dask.utils import ( |
44 | | - get_default_shuffle_algorithm, |
45 | | - parse_timedelta, |
46 | | - stringify, |
47 | | - tmpfile, |
48 | | -) |
| 43 | +from dask.utils import get_default_shuffle_method, parse_timedelta, stringify, tmpfile |
49 | 44 |
|
50 | 45 | from distributed import ( |
51 | 46 | CancelledError, |
@@ -3229,25 +3224,25 @@ def test_default_get(loop_in_thread): |
3229 | 3224 | # These may change in the future but the selection below should not |
3230 | 3225 | distributed_default = "p2p" if has_pyarrow else "tasks" |
3231 | 3226 | local_default = "disk" |
3232 | | - assert get_default_shuffle_algorithm() == local_default |
| 3227 | + assert get_default_shuffle_method() == local_default |
3233 | 3228 | with Client(s["address"], set_as_default=True, loop=loop) as c: |
3234 | 3229 | assert dask.base.get_scheduler() == c.get |
3235 | | - assert get_default_shuffle_algorithm() == distributed_default |
| 3230 | + assert get_default_shuffle_method() == distributed_default |
3236 | 3231 |
|
3237 | 3232 | assert dask.base.get_scheduler() == pre_get |
3238 | | - assert get_default_shuffle_algorithm() == local_default |
| 3233 | + assert get_default_shuffle_method() == local_default |
3239 | 3234 |
|
3240 | 3235 | c = Client(s["address"], set_as_default=False, loop=loop) |
3241 | 3236 | assert dask.base.get_scheduler() == pre_get |
3242 | | - assert get_default_shuffle_algorithm() == local_default |
| 3237 | + assert get_default_shuffle_method() == local_default |
3243 | 3238 | c.close() |
3244 | 3239 |
|
3245 | 3240 | c = Client(s["address"], set_as_default=True, loop=loop) |
3246 | | - assert get_default_shuffle_algorithm() == distributed_default |
| 3241 | + assert get_default_shuffle_method() == distributed_default |
3247 | 3242 | assert dask.base.get_scheduler() == c.get |
3248 | 3243 | c.close() |
3249 | 3244 | assert dask.base.get_scheduler() == pre_get |
3250 | | - assert get_default_shuffle_algorithm() == local_default |
| 3245 | + assert get_default_shuffle_method() == local_default |
3251 | 3246 |
|
3252 | 3247 | with Client(s["address"], loop=loop) as c: |
3253 | 3248 | assert dask.base.get_scheduler() == c.get |
|
0 commit comments