[misc] format InterleaveThinker workflow helpers
This commit is contained in:
@@ -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),
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user