From 2f93a6c9b87e85439ddc21b44ba41dc6d2111fb0 Mon Sep 17 00:00:00 2001 From: Tung-Cheng Wu Date: Wed, 1 Oct 2025 22:32:18 -0400 Subject: [PATCH] correct checker to be as same as argument's help message --- data_utils/process_waymo_files.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/data_utils/process_waymo_files.py b/data_utils/process_waymo_files.py index 8c405c913..6428099b1 100644 --- a/data_utils/process_waymo_files.py +++ b/data_utils/process_waymo_files.py @@ -624,7 +624,7 @@ def process_data(args): if args.dataset == "all": datasets = ["training", "validation", "testing"] - elif args.dataset == "train": + elif args.dataset == "training": datasets = ["training"] elif args.dataset == "validation": datasets = ["validation"] @@ -632,7 +632,7 @@ def process_data(args): datasets = ["testing"] else: raise ValueError( - "Invalid dataset name. Must be one of: 'all', 'train', 'validation', or 'testing'" + "Invalid dataset name. Must be one of: 'all', 'training', 'validation', or 'testing'" ) for dataset in datasets: