Validate that inputdir is a filtered raw subdir

swiss-ai · Aug 5, 2024 · 6421dd5 · 6421dd5
1 parent 7c98e82
commit 6421dd5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/pseudolabeling/v2d_to_metadata.py b/pseudolabeling/v2d_to_metadata.py
@@ -94,6 +94,9 @@ def process_json_file(json_file_path, output_dir, dataset):
 
 
 def main(args):
+    if "filtered_raw" not in args.input_dir:
+        raise ValueError(f"Expected input dir to be a subdir of `filtered_raw/`, instead received {args.input_dir}.")
+
     output_dir = (
         args.output_dir
         if args.output_dir is not None

diff --git a/pseudolabeling/v2d_to_transcript.py b/pseudolabeling/v2d_to_transcript.py
@@ -103,6 +103,9 @@ def process_json_file(json_file_path, output_dir):
 
 
 def main(args):
+    if "filtered_raw" not in args.input_dir:
+        raise ValueError(f"Expected input dir to be a subdir of `filtered_raw/`, instead received {args.input_dir}.")
+
     current_folder = os.path.join(args.input_dir, args.whisper_dir)
     output_dir = (
         args.output_dir