From d0200d1dbead3d855e0332a9921e06470c3e45ec Mon Sep 17 00:00:00 2001 From: joesharratt1229 Date: Sun, 16 Feb 2025 15:56:58 +0000 Subject: [PATCH] formatted answer as str --- reasoning_gym/logic/aiw.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/reasoning_gym/logic/aiw.py b/reasoning_gym/logic/aiw.py index 7130a11d..00448280 100644 --- a/reasoning_gym/logic/aiw.py +++ b/reasoning_gym/logic/aiw.py @@ -187,7 +187,7 @@ class AliceInWonderlandDataset(ProceduralDataset): num_female_colleagues_bob_circle=num_female_colleagues_bob_circle, ) - return {"question": question, "answer": answer, "metadata": {"task_type": task_type.value}} + return {"question": question, "answer": str(answer), "metadata": {"task_type": task_type.value}} def __getitem__(self, idx: int) -> dict: rng = Random(self.seed + idx)