From 5b2317ccb2f332af57b0a813bacaf80fdf734739 Mon Sep 17 00:00:00 2001
From: Taiwei Shi <39599206+maksimstw@users.noreply.github.com>
Date: Wed, 12 Feb 2025 22:16:10 -0800
Subject: [PATCH] Update __init__.py

allowing extra_info to be passed in for more advance compute_score
---
 verl/utils/reward_score/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/verl/utils/reward_score/__init__.py b/verl/utils/reward_score/__init__.py
index f53662c1..ee8fbb60 100644
--- 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)