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

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-06-27 07:50:11 +00:00
commit 5b5a863445
2 changed files with 9 additions and 5 deletions

View file

@ -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()

View file

@ -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