Update __init__.py

allowing extra_info to be passed in for more advance compute_score
volcengine · Feb 13, 2025 · 5b2317c · 5b2317c
1 parent eada848
commit 5b2317c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/verl/utils/reward_score/__init__.py b/verl/utils/reward_score/__init__.py
@@ -14,7 +14,7 @@
 # from . import gsm8k, math, prime_math, prime_code
 
 
-def _default_compute_score(data_source, solution_str, ground_truth):
+def _default_compute_score(data_source, solution_str, ground_truth, extra_info):
     if data_source == 'openai/gsm8k':
         from . import gsm8k
         res = gsm8k.compute_score(solution_str, ground_truth)