From 62d05429c73dc91569bd6a4d16276f5ca578229d Mon Sep 17 00:00:00 2001 From: whatchang Date: Mon, 12 Feb 2024 13:42:04 +0900 Subject: [PATCH] modify value_outputs_unwrap method --- src/tot/tasks/game24.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/tot/tasks/game24.py b/src/tot/tasks/game24.py index c58f090..77b0cc3 100644 --- a/src/tot/tasks/game24.py +++ b/src/tot/tasks/game24.py @@ -88,5 +88,5 @@ class Game24Task(Task): return 0 value_names = [_.split('\n')[-1] for _ in value_outputs] value_map = {'impossible': 0.001, 'likely': 1, 'sure': 20} # TODO: ad hoc - value = sum(value * value_names.count(name) for name, value in value_map.items()) + value = sum(value * target_value.count(name) for name, value in value_map.items() for target_value in value_names) return value \ No newline at end of file