[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

studio/backend/hub/services/download_lifecycle.py

@@ -277,9 +277,7 @@ def finalize_worker_exit(
 
 
 def _set_job_transport(
-    registry: download_registry.DownloadRegistry,
-    key: str,
-    transport: str,
+    registry: download_registry.DownloadRegistry, key: str, transport: str
 ) -> None:
     with registry._lock:
         metadata = registry._metadata.get(key)
@@ -375,7 +373,9 @@ def _try_http_retry(
         replace_active = True,
     )
     if not claimed:
-        logger.debug("%s XET retry claim rejected for %s; another job took the slot", log_prefix, label)
+        logger.debug(
+            "%s XET retry claim rejected for %s; another job took the slot", log_prefix, label
+        )
         _set_retry_failure_state(
             registry,
             key,
@@ -492,7 +492,8 @@ def _watch() -> None:
                 transport == download_registry.TRANSPORT_XET
                 and download_registry.download_transport_unavailable_reason(
                     download_registry.TRANSPORT_HTTP
-                ) is None
+                )
+                is None
             )
             state = finalize_worker_exit(
                 registry,
@@ -511,10 +512,7 @@ def _watch() -> None:
             # HTTP is available, attempt one automatic retry over HTTP.  The
             # transport check is the recursion guard: an HTTP worker that errors
             # never satisfies `transport == TRANSPORT_XET`, so it stays terminal.
-            if (
-                can_retry_http
-                and state == "error"
-            ):
+            if can_retry_http and state == "error":
                 _try_http_retry(
                     registry,
                     key,

studio/backend/hub/tests/test_download_lifecycle.py

@@ -17,7 +17,7 @@ def _set_xet_reason(monkeypatch, reason):
     )
 
 
-def _make_proc(rc, stderr=b""):
+def _make_proc(rc, stderr = b""):
     class _Proc:
         pid = 4242
 
@@ -63,7 +63,9 @@ def test_resolve_effective_use_xet_downgrades_when_xet_unavailable(monkeypatch):
     assert download_lifecycle.resolve_effective_use_xet(True) is False
 
 
-def test_download_watcher_retries_xet_failure_over_http_for_model_and_dataset(monkeypatch, tmp_path):
+def test_download_watcher_retries_xet_failure_over_http_for_model_and_dataset(
+    monkeypatch, tmp_path
+):
     monkeypatch.setattr(state_dir, "cache_root", lambda: tmp_path / "state")
     monkeypatch.setattr(download_lifecycle.threading, "Thread", _ImmediateThread)
     real_register_worker = download_lifecycle.register_worker
@@ -127,7 +129,13 @@ def fake_completed_blob_bytes(*args):
             baseline_calls.append(args)
             return retry_baseline_bytes
 
-        def fake_spawn_worker(args, hf_token, *, use_xet, protected_blob_hashes = None):
+        def fake_spawn_worker(
+            args,
+            hf_token,
+            *,
+            use_xet,
+            protected_blob_hashes = None,
+        ):
             assert registry.get_job(key).state == "running"
             spawned.append((args, use_xet, protected_blob_hashes))
             return _make_proc(0, b"http retry")
@@ -155,7 +163,9 @@ def fake_register_worker(*_args, **kwargs):
         )
 
         assert spawned == [(expected_args, False, None)]
-        assert retry_registers and retry_registers[0]["transport"] == download_registry.TRANSPORT_HTTP
+        assert (
+            retry_registers and retry_registers[0]["transport"] == download_registry.TRANSPORT_HTTP
+        )
         metadata = registry.get_job_metadata(key)
         assert metadata is not None
         assert metadata.transport == download_registry.TRANSPORT_HTTP
@@ -205,9 +215,7 @@ def fake_register_worker(*_args, **_kwargs):
     assert registry.get_job(key).state == "error"
 
 
-def test_download_watcher_restores_xet_transport_when_http_retry_spawn_fails(
-    monkeypatch, tmp_path
-):
+def test_download_watcher_restores_xet_transport_when_http_retry_spawn_fails(monkeypatch, tmp_path):
     monkeypatch.setattr(state_dir, "cache_root", lambda: tmp_path / "state")
     monkeypatch.setattr(download_lifecycle.threading, "Thread", _ImmediateThread)
     real_register_worker = download_lifecycle.register_worker

studio/backend/hub/utils/download_registry.py

@@ -849,22 +849,16 @@ def set_job(
                 self._jobs[key] = DownloadState(state, error)
 
     def set_error_unless_cancelled(
-        self,
-        key: str,
-        error: str,
+        self, key: str, error: str
     ) -> tuple[JobState, Optional[DownloadMetadata]]:
         key = normalize_job_key(key)
         with self._lock:
             current = self._jobs.get(key, DownloadState("idle")).state
             has_pending_cancel = key in self._pending_cancel
             pending_generation = self._pending_cancel.get(key)
             metadata = self._metadata.get(key)
-            should_cancel = (
-                current == "cancelling"
-                or (
-                    has_pending_cancel
-                    and self._generation_matches_locked(key, pending_generation)
-                )
+            should_cancel = current == "cancelling" or (
+                has_pending_cancel and self._generation_matches_locked(key, pending_generation)
             )
             terminal_state: JobState = "cancelled" if should_cancel else "error"
             self._put_terminal_job_locked(