add warning when no transcriptions.csv found

qiuqiao · Mar 3, 2024 · 51a869d · 51a869d
1 parent c6bf62e
commit 51a869d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/binarize.py b/binarize.py
@@ -1,4 +1,5 @@
 import pathlib
+import warnings
 
 import click
 import h5py
@@ -320,6 +321,8 @@ def get_meta_data(self, data_folder, vocab):
             for i in path.rglob("transcriptions.csv")
             if i.name == "transcriptions.csv"
         ]
+        if len(trans_path_list) <= 0:
+            warnings.warn(f"No transcriptions.csv found in {data_folder}.")
 
         print("Loading metadata...")
         meta_data_df = pd.DataFrame()