[misc] format InterleaveThinker workflow helpers

This commit is contained in:
Mac Lee
2026-06-21 19:03:31 +00:00
parent 11d55fb5a2
commit 91d8fb85e6
2 changed files with 6 additions and 12 deletions
@@ -95,9 +95,7 @@ def build_planner(config: InterleavePlannerConfig) -> PlannerProvider:
if config.kind == "single_prompt":
return SinglePromptPlanner(max_attempts=config.max_attempts_per_step)
model = InterleaveThinkerPlannerModel(
**_actor_model_kwargs(config),
)
model = InterleaveThinkerPlannerModel(**_actor_model_kwargs(config), )
return InterleaveThinkerPlannerProvider(
model,
num_generations=config.num_generations,
@@ -114,9 +112,7 @@ def build_critic(config: InterleaveCriticConfig) -> CriticProvider | None:
if config.kind == "accept_all":
return AcceptAllCritic()
model = InterleaveThinkerCriticModel(
**_actor_model_kwargs(config),
)
model = InterleaveThinkerCriticModel(**_actor_model_kwargs(config), )
return InterleaveThinkerCriticProvider(
model,
num_generations=config.num_generations,
@@ -27,12 +27,10 @@ def trace_to_dict(
trace.final_image,
include_images=include_images,
),
"attempts": [
_attempt_to_dict(
attempt,
include_images=include_images,
) for attempt in trace.attempts
],
"attempts": [_attempt_to_dict(
attempt,
include_images=include_images,
) for attempt in trace.attempts],
"metadata": dict(trace.metadata),
}