From 535371a5b85c16df815ac4b8749263dc4322ff3f Mon Sep 17 00:00:00 2001 From: Khalil Meftah Date: Mon, 13 Jul 2026 14:58:53 +0200 Subject: [PATCH] feat(aggregate): initialize language type columns in aggregate_data function --- src/lerobot/datasets/aggregate.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/lerobot/datasets/aggregate.py b/src/lerobot/datasets/aggregate.py index 0fd996c24..c9f6a34ff 100644 --- a/src/lerobot/datasets/aggregate.py +++ b/src/lerobot/datasets/aggregate.py @@ -569,7 +569,9 @@ def aggregate_data(src_meta, dst_meta, data_idx, data_files_size_in_mb, chunk_si if dtype in ("video", "image"): continue n_rows = len(df) - if dtype == "bool": + if dtype == "language": + df[col_name] = [[] for _ in range(n_rows)] + elif dtype == "bool": df[col_name] = False elif dtype in ("float32", "float64"): df[col_name] = 0.0