fix a typo on COLUMNS definition

2025-09-22 17:33:28 +08:00
parent 7b4d383611
commit 5dcd1ca503
1 changed files with 1 additions and 1 deletions
--- 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'
    ]