Skip to content

Commit 942a2f3

Browse files
abrichrclaude
andauthored
fix: use ResetConfig for RLEnvironment.reset() in validation script (#56)
The validation script called env.reset(task_id=...) but the actual API is env.reset(config=ResetConfig(task_id=...)). This caused Phase 2 to fail with TypeError. Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent 1b8ae78 commit 942a2f3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

scripts/validate_grpo_waa.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,12 +81,12 @@ def phase2_single_rollout(
8181
)
8282

8383
from openadapt_evals.adapters.base import BenchmarkAction
84-
from openadapt_evals.adapters.rl_env import RLEnvironment
84+
from openadapt_evals.adapters.rl_env import RLEnvironment, ResetConfig
8585

8686
env = RLEnvironment(adapter)
8787

8888
# Reset
89-
obs = env.reset(task_id=task_id)
89+
obs = env.reset(config=ResetConfig(task_id=task_id))
9090
if obs is None or obs.screenshot is None:
9191
logger.error(" Reset returned no observation or screenshot")
9292
return False

0 commit comments

Comments
 (0)