From f97ef32100fdeb0b4ee130193d156fade7c41ac0 Mon Sep 17 00:00:00 2001
From: pyr0ball <pyroballpcs@gmail.com>
Date: Fri, 27 Feb 2026 14:54:31 -0800
Subject: [PATCH] fix: RerankerAdapter falls back to label name when no
 LABEL_DESCRIPTIONS entry

---
 scripts/classifier_adapters.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/classifier_adapters.py b/scripts/classifier_adapters.py
index e6020e2..2817078 100644
--- a/scripts/classifier_adapters.py
+++ b/scripts/classifier_adapters.py
@@ -258,6 +258,6 @@ class RerankerAdapter(ClassifierAdapter):
         if self._reranker is None:
             self.load()
         text = f"Subject: {subject}\n\n{body[:600]}"
-        pairs = [[text, LABEL_DESCRIPTIONS[label]] for label in LABELS]
+        pairs = [[text, LABEL_DESCRIPTIONS.get(label, label.replace("_", " "))] for label in LABELS]
         scores: list[float] = self._reranker.compute_score(pairs, normalize=True)
         return LABELS[scores.index(max(scores))]