diff --git a/studio/backend/hub/utils/download_registry.py b/studio/backend/hub/utils/download_registry.py index e459562f98..5e1d241aa6 100644 --- a/studio/backend/hub/utils/download_registry.py +++ b/studio/backend/hub/utils/download_registry.py @@ -1046,7 +1046,11 @@ def _resumable_blob_hashes(entry: Path) -> set[str]: def _manifest_hash_split( - repo_type: str, repo_id: str, variant: Optional[str], *, root: Optional[Path] = None + repo_type: str, + repo_id: str, + variant: Optional[str], + *, + root: Optional[Path] = None, ) -> tuple[set[str], set[str]]: """``(main, companion)`` blob hashes from the variant's manifest, split the way the worker splits them when it asks for a purge. Empty pair when either step cannot answer, which diff --git a/studio/backend/tests/test_partial_resume_verdict.py b/studio/backend/tests/test_partial_resume_verdict.py index 6ff4f8b6d8..18b5476e5c 100644 --- a/studio/backend/tests/test_partial_resume_verdict.py +++ b/studio/backend/tests/test_partial_resume_verdict.py @@ -261,7 +261,12 @@ def two_roots(monkeypatch, tmp_path): return active / "models--Org--Model", previous / "models--Org--Model" -def _record_in(entry: Path, transport: str, variant, blob: str = BLOB): +def _record_in( + entry: Path, + transport: str, + variant, + blob: str = BLOB, +): download_manifest.write_manifest( "model", "Org/Model",