From ed1040dd27d190e5bbc9079197a7f3c0cea5ed57 Mon Sep 17 00:00:00 2001 From: Simon Adorf Date: Tue, 12 Sep 2023 11:38:41 -0700 Subject: [PATCH] Fix _extract_partitions function. --- python/cuml/dask/common/part_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/cuml/dask/common/part_utils.py b/python/cuml/dask/common/part_utils.py index 39466bf075..a6aa892a76 100644 --- a/python/cuml/dask/common/part_utils.py +++ b/python/cuml/dask/common/part_utils.py @@ -160,7 +160,7 @@ def _extract_partitions(dask_obj, client=None): yield wait(parts) - key_to_part = [(str(part.key), part) for part in parts] + key_to_part = [(part.key, part) for part in parts] who_has = yield client.who_has(parts) raise gen.Return(