From 305699bbbdde9b555ecd88c4bedc0b312a268a53 Mon Sep 17 00:00:00 2001 From: Charles Stern <62192187+cisaacstern@users.noreply.github.com> Date: Thu, 12 Oct 2023 16:42:52 -0700 Subject: [PATCH] cast dask_visitor.bags.values() to list in dask.optimize call --- sdks/python/apache_beam/runners/dask/dask_runner.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/sdks/python/apache_beam/runners/dask/dask_runner.py b/sdks/python/apache_beam/runners/dask/dask_runner.py index f11d139c0c94..81e263d82626 100644 --- a/sdks/python/apache_beam/runners/dask/dask_runner.py +++ b/sdks/python/apache_beam/runners/dask/dask_runner.py @@ -183,7 +183,6 @@ def run_pipeline(self, pipeline, options): dask_visitor = self.to_dask_bag_visitor() pipeline.visit(dask_visitor) - - opt_graph = dask.optimize(dask_visitor.bags.values()) + opt_graph = dask.optimize(list(dask_visitor.bags.values())) futures = client.compute(opt_graph) return DaskRunnerResult(client, futures)