sebastien
/
Copies
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

read names from parent directory

master
Sébastien Miquel 2026-03-31 11:17:08 +02:00
parent 3d26a8ec01
commit 133be2a84b
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
gemini_for_labels.py
View File

@ -210,7 +210,11 @@ for path_str in args.input_paths:
labels_txt = (INPUT_DIR / "labels").read_text() labels_txt = (INPUT_DIR / "labels").read_text()
valid_labels_set = set(line.strip() for line in labels_txt.splitlines() if line.strip()) 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 = set(line.strip() for line in names_txt.splitlines() if line.strip())
valid_names_set.add("Unknown") valid_names_set.add("Unknown")
valid_names_set.add("Continued") valid_names_set.add("Continued")