From cee2bf6ed2f9701cd73058e1054f5386b70c65fe Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 10 Jul 2026 09:20:07 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- studio/backend/core/inference/video.py | 5 ++--- studio/backend/tests/test_video_routes.py | 4 +--- 2 files changed, 3 insertions(+), 6 deletions(-) diff --git a/studio/backend/core/inference/video.py b/studio/backend/core/inference/video.py index 248764aa87..05d9825df5 100644 --- a/studio/backend/core/inference/video.py +++ b/studio/backend/core/inference/video.py @@ -1594,9 +1594,7 @@ class VideoBackend: cancel_event = cancel_event, error = "Failed to save the generated video." ) return - self._finish_generate_job( - cancel_event = cancel_event, video = record, total = result["steps"] - ) + self._finish_generate_job(cancel_event = cancel_event, video = record, total = result["steps"]) def _finish_generate_job( self, @@ -1653,6 +1651,7 @@ class VideoBackend: cancel_event: Optional[threading.Event] = None, ) -> dict[str, Any]: import torch + # begin_generate passes the event it already registered (so a cancel in the # spawn window is honoured); a direct call makes its own. cancel = cancel_event if cancel_event is not None else threading.Event() diff --git a/studio/backend/tests/test_video_routes.py b/studio/backend/tests/test_video_routes.py index 6c207bfafa..a82daf6d6a 100644 --- a/studio/backend/tests/test_video_routes.py +++ b/studio/backend/tests/test_video_routes.py @@ -321,9 +321,7 @@ def test_load_threads_transformer_quant_and_guidance_2(client): kwargs = video_module.get_video_backend().last_load_kwargs assert kwargs.get("transformer_quant") == "fp8" - video = _generate_and_wait( - client, {"prompt": "a sloth", "guidance": 5.0, "guidance_2": 3.0} - ) + video = _generate_and_wait(client, {"prompt": "a sloth", "guidance": 5.0, "guidance_2": 3.0}) assert video["guidance"] == 5.0 and video["guidance_2"] == 3.0