Change weight to float

macrocosm-os · Jan 22, 2024 · 4685065 · 4685065
1 parent 5885925
commit 4685065
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/prompting/tasks/date_qa.py b/prompting/tasks/date_qa.py
@@ -6,7 +6,7 @@
 @dataclass
 class DateQuestionAnsweringTask(Task):
     reward_definition = [
-        dict(name="date", weight=1),
+        dict(name="date", weight=1.0),
     ]
     penalty_definition = []