diff --git a/sdks/python/apache_beam/runners/dask/transform_evaluator.py b/sdks/python/apache_beam/runners/dask/transform_evaluator.py index ea2df52192db..ac557fb0c700 100644 --- a/sdks/python/apache_beam/runners/dask/transform_evaluator.py +++ b/sdks/python/apache_beam/runners/dask/transform_evaluator.py @@ -87,10 +87,7 @@ class DaskBagWindowedIterator: window_fn: WindowFn def __iter__(self): - # FIXME(cisaacstern): list() is likely inefficient, since it presumably - # materializes the full result before iterating over it. doing this for - # now as a proof-of-concept. can we can generate results incrementally? - for result in list(self.bag): + for result in iter(self.bag): yield get_windowed_value(result, self.window_fn)