diff --git a/src/dask_awkward/lib/optimize.py b/src/dask_awkward/lib/optimize.py index 51660d1d..f2dbe201 100644 --- a/src/dask_awkward/lib/optimize.py +++ b/src/dask_awkward/lib/optimize.py @@ -389,8 +389,8 @@ def _get_column_reports(dsk: HighLevelGraph) -> dict[str, Any]: try: for layer in hlg.layers.values(): layer.__dict__.pop("_cached_dict", None) - for outlayerkey in leaf_layers_keys: - out = get_sync(hlg, outlayerkey) + results = get_sync(hlg, leaf_layers_keys) + for out in results: if isinstance(out, (ak.Array, ak.Record)): out.layout._touch_data(recursive=True) except Exception as err: