From 133be2a84b2e7ae3146090a33d59111a456ee5da Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?S=C3=A9bastien=20Miquel?= Date: Tue, 31 Mar 2026 11:17:08 +0200 Subject: [PATCH] read names from parent directory --- gemini_for_labels.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/gemini_for_labels.py b/gemini_for_labels.py index ffab75e..859872b 100644 --- a/gemini_for_labels.py +++ b/gemini_for_labels.py @@ -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")