Fix reward_fn sandbox signature
Browse files- 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"]
|
| 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}")
|