From 5b5a8634452eaafdcd2a137642154a1257df85ff Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sat, 27 Jun 2026 07:50:11 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- studio/backend/loggers/progress.py | 7 +++++-- studio/backend/tests/test_progress_throttle.py | 7 ++++--- 2 files changed, 9 insertions(+), 5 deletions(-) diff --git a/studio/backend/loggers/progress.py b/studio/backend/loggers/progress.py index 8deee865fa..0a136532d9 100644 --- a/studio/backend/loggers/progress.py +++ b/studio/backend/loggers/progress.py @@ -24,7 +24,6 @@ def _interval_default() -> float: def _verbose() -> bool: from loggers.config import logs_verbose - return logs_verbose() @@ -40,7 +39,11 @@ class ProgressThrottle: self._last_msg: dict = {} self._guard = threading.Lock() - def should_log(self, key, message: str = "") -> bool: + def should_log( + self, + key, + message: str = "", + ) -> bool: if self._interval <= 0 or _verbose(): return True now = time.monotonic() diff --git a/studio/backend/tests/test_progress_throttle.py b/studio/backend/tests/test_progress_throttle.py index 1f48c5a911..323a0ce7d6 100644 --- a/studio/backend/tests/test_progress_throttle.py +++ b/studio/backend/tests/test_progress_throttle.py @@ -17,7 +17,7 @@ def _not_verbose(monkeypatch): def test_first_message_logs_then_identical_repeats_throttle(): t = ProgressThrottle(interval_s = 100) # large window: only the first logs k = "load" - assert t.should_log(k, "Downloading...") is True # first + assert t.should_log(k, "Downloading...") is True # first assert t.should_log(k, "Downloading...") is False # identical repeat within window assert t.should_log(k, "Downloading...") is False @@ -27,11 +27,12 @@ def test_message_change_always_logs(): k = "load" assert t.should_log(k, "Loading model...") is True assert t.should_log(k, "Loading model...") is False - assert t.should_log(k, "Importing Unsloth...") is True # phase change logs + assert t.should_log(k, "Importing Unsloth...") is True # phase change logs def test_heartbeat_emits_after_interval(): import time + t = ProgressThrottle(interval_s = 0.05) k = "load" assert t.should_log(k, "same") is True @@ -73,5 +74,5 @@ def test_verbose_disables_throttle(monkeypatch): def test_distinct_keys_are_independent(): t = ProgressThrottle(interval_s = 100) assert t.should_log("a", "m") is True - assert t.should_log("b", "m") is True # different key, independent + assert t.should_log("b", "m") is True # different key, independent assert t.should_log("a", "m") is False