read names from parent directory

2026-03-31 11:17:08 +02:00
parent 3d26a8ec01
commit 133be2a84b
1 changed files with 5 additions and 1 deletions
@@ -210,7 +210,11 @@ for path_str in args.input_paths:

 labels_txt = (INPUT_DIR / "labels").read_text()
 valid_labels_set = set(line.strip() for line in labels_txt.splitlines() if line.strip())
-names_txt = (INPUT_DIR / "names").read_text()
+names_path = (INPUT_DIR / "names")
+if !os.path.exists(names_path):
+    names_path = Path("names")
+names_txt = names_path.read_text()
+
 valid_names_set = set(line.strip() for line in names_txt.splitlines() if line.strip())
 valid_names_set.add("Unknown")
 valid_names_set.add("Continued")