small fix for train_raw & train_refined

yfyeung · Jun 28, 2024 · 83aef17 · 83aef17
1 parent 1d721b4
commit 83aef17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lhotse/recipes/gigaspeech2.py b/lhotse/recipes/gigaspeech2.py
@@ -97,7 +97,7 @@ def _prepare_subset(
     :return: the RecodingSet and SupervisionSet for train and valid.
     """
     lang_dir = Path(lang_dir)
-    part_dir = lang_dir / part
+    part_dir = lang_dir / part.replace("_raw", "").replace("_refined", "")
     tsv_path = lang_dir / f"{part}.tsv"
 
     audio_infos = []