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)