From 5dcd1ca50391f1bb5053ffa42c805f382920d93c Mon Sep 17 00:00:00 2001
From: yuchen-x <rd_xyc@unitree.com>
Date: Mon, 22 Sep 2025 17:33:28 +0800
Subject: [PATCH] fix a typo on COLUMNS definition

---
 prepare_data/prepare_training_data.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/prepare_data/prepare_training_data.py b/prepare_data/prepare_training_data.py
index 71cb917..3d899e3 100644
--- a/prepare_data/prepare_training_data.py
+++ b/prepare_data/prepare_training_data.py
@@ -74,7 +74,7 @@ def main(args):
 
     csv_file = target_dir / f"{args.dataset_name}.csv"
     COLUMNS = [
-        'videoid', 'contentUrl', 'duration', 'data_dir', 'name',
+        'videoid', 'contentUrl', 'duration', 'data_dir', 'instruction',
         'dynamic_confidence', 'dynamic_wording', 'dynamic_source_category',
         'embodiment'
     ]