Circuit-Forge/peregrine
2
0
Fork 0
Code Issues 22 Pull requests 2 Projects Releases 6 Packages Wiki Activity Actions

fix: RerankerAdapter falls back to label name when no LABEL_DESCRIPTIONS entry

Browse source
This commit is contained in:
pyr0ball 2026-02-27 14:54:31 -08:00
parent 23828520f0
commit 9fe9c6234d
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
scripts/classifier_adapters.py
View file

@ -258,6 +258,6 @@ class RerankerAdapter(ClassifierAdapter):
if self._reranker is None: if self._reranker is None:
self.load() self.load()
text = f"Subject: {subject}\n\n{body[:600]}" 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) scores: list[float] = self._reranker.compute_score(pairs, normalize=True)
return LABELS[scores.index(max(scores))] return LABELS[scores.index(max(scores))]