diff --git a/fastvideo/workflows/interleave_thinker/runner.py b/fastvideo/workflows/interleave_thinker/runner.py index 83d651aaa..69aa30898 100644 --- a/fastvideo/workflows/interleave_thinker/runner.py +++ b/fastvideo/workflows/interleave_thinker/runner.py @@ -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, diff --git a/fastvideo/workflows/interleave_thinker/trace.py b/fastvideo/workflows/interleave_thinker/trace.py index e09cf3a00..86f34b31e 100644 --- a/fastvideo/workflows/interleave_thinker/trace.py +++ b/fastvideo/workflows/interleave_thinker/trace.py @@ -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), }