vasiuuu commited on
Commit
6718f28
·
1 Parent(s): 6bb304d

Fix reward_fn sandbox signature

Browse files
Files changed (1) hide show
  1. trainer/reward_fn.py +1 -1
trainer/reward_fn.py CHANGED
@@ -19,7 +19,7 @@ def codeforge_reward_fn(prompts, completions, *, task_meta, **kwargs):
19
  sandbox_files = {**files, **meta["hidden_tests"]}
20
  try:
21
  # CodeForge run_sandbox signature:
22
- result = run_sandbox(files=sandbox_files, tools=meta["tools"], timeout=10.0)
23
  sandbox_score = result.composite_score
24
  except Exception as e:
25
  print(f"Sandbox error: {e}")
 
19
  sandbox_files = {**files, **meta["hidden_tests"]}
20
  try:
21
  # CodeForge run_sandbox signature:
22
+ result = run_sandbox(files=sandbox_files, tools=meta["tools"])
23
  sandbox_score = result.composite_score
24
  except Exception as e:
25
  print(f"Sandbox error: {e}")