Skip to content

Commit b2123ad

Browse files
authored
correct checker to be as same as argument's help message (#525)
1 parent b2e575f commit b2123ad

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

data_utils/process_waymo_files.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -624,15 +624,15 @@ def process_data(args):
624624

625625
if args.dataset == "all":
626626
datasets = ["training", "validation", "testing"]
627-
elif args.dataset == "train":
627+
elif args.dataset == "training":
628628
datasets = ["training"]
629629
elif args.dataset == "validation":
630630
datasets = ["validation"]
631631
elif args.dataset == "testing":
632632
datasets = ["testing"]
633633
else:
634634
raise ValueError(
635-
"Invalid dataset name. Must be one of: 'all', 'train', 'validation', or 'testing'"
635+
"Invalid dataset name. Must be one of: 'all', 'training', 'validation', or 'testing'"
636636
)
637637

638638
for dataset in datasets:

0 commit comments

Comments
 (0)