Skip to content

Commit f29110d

Browse files
committed
export ep params
1 parent 8d72105 commit f29110d

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

eval_protocol/pytest/evaluation_test.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -704,11 +704,22 @@ async def _collect_result(config, lst):
704704
)
705705
pytest_wrapper = pytest.mark.asyncio(pytest_wrapper)
706706

707+
ep_params: dict[str, Any] = {
708+
"rollout_processor": rollout_processor,
709+
"server_script_path": server_script_path,
710+
"mcp_config_path": mcp_config_path,
711+
"rollout_processor_kwargs": rollout_processor_kwargs,
712+
"mode": mode,
713+
}
714+
715+
print(f"ep_params: {ep_params}")
716+
707717
# Create the dual mode wrapper
708718
dual_mode_wrapper = create_dual_mode_wrapper(
709719
test_func, mode, max_concurrent_rollouts, max_concurrent_evaluations, pytest_wrapper
710720
)
711721

722+
setattr(dual_mode_wrapper, "__ep_params__", ep_params)
712723
return dual_mode_wrapper # pyright: ignore[reportReturnType, reportUnknownVariableType]
713724

714725
return decorator

0 commit comments

Comments
 (0)