diff --git a/studio/backend/core/inference/video.py b/studio/backend/core/inference/video.py index c5828a7793..f4110d0c74 100644 --- a/studio/backend/core/inference/video.py +++ b/studio/backend/core/inference/video.py @@ -756,7 +756,11 @@ class VideoBackend: entries: dict[str, dict[str, Any]] = {} total = 0 - def add(repo: str, files: list[tuple[str, int]], gguf: Optional[str] = None) -> int: + def add( + repo: str, + files: list[tuple[str, int]], + gguf: Optional[str] = None, + ) -> int: if not files: return 0 entry = entries.setdefault( diff --git a/studio/backend/tests/test_video_backend.py b/studio/backend/tests/test_video_backend.py index 3a90a2248e..4fb3c74d69 100644 --- a/studio/backend/tests/test_video_backend.py +++ b/studio/backend/tests/test_video_backend.py @@ -1861,7 +1861,12 @@ _LTX23_REPO_SIBLINGS = [ def _plan_api(monkeypatch, repos): class _Api: - def model_info(self, repo_id, files_metadata = False, token = None): + def model_info( + self, + repo_id, + files_metadata = False, + token = None, + ): return _PlanInfo(repos[repo_id]) monkeypatch.setattr("huggingface_hub.HfApi", lambda *a, **k: _Api()) @@ -1894,9 +1899,7 @@ def test_download_plan_narrows_an_ltx23_pick_and_stages_its_extras(monkeypatch): assert ckpt["gguf_filename"] == "ltx-2.3-22b-distilled.gguf" assert "vae/ltx-2.3-22b-distilled_video_vae.safetensors" in ckpt["files"] assert "vae/ltx-2.3-22b-distilled_audio_vae.safetensors" in ckpt["files"] - assert ( - "text_encoders/ltx-2.3-22b-distilled_embeddings_connectors.safetensors" in ckpt["files"] - ) + assert "text_encoders/ltx-2.3-22b-distilled_embeddings_connectors.safetensors" in ckpt["files"] # The other variant's companions are not this checkpoint's. assert "vae/ltx-2.3-22b-dev_video_vae.safetensors" not in ckpt["files"]