ensure reward is float

This commit is contained in:
Andreas Koepf 2025-02-16 16:27:12 +01:00
parent 66ddb41bbd
commit 6fae41a6e1
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: