cast dask_visitor.bags.values() to list in dask.optimize call

apache · Oct 12, 2023 · 305699b · 305699b
1 parent 32ee6e1
commit 305699b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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)