diff --git a/distributed/client.py b/distributed/client.py index 0b4214f2304..1e7b96b2aa5 100644 --- a/distributed/client.py +++ b/distributed/client.py @@ -3595,7 +3595,7 @@ def persist( result = [] for collection in collections: graph = ( - collection.finalize_compute().__dask_graph__() + collection.__dask_graph__() # Currently optimize should be called on client side to # fix dask keys .optimize()