ensure reward is float

This commit is contained in:
Andreas Koepf 2025-02-16 16:27:12 +01:00
parent c858d1f236
commit 4c47b7966f
3 changed files with 5 additions and 5 deletions

View file

@ -93,7 +93,7 @@ class SentenceReorderingDataset(ProceduralDataset):
}
def score_answer(self, answer: Optional[str], entry: Dict[str, Any]) -> float:
reward = 0
reward = 0.0
expected_answer = entry["answer"]
if answer is not None:
try: