ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-03-12 09:01:56 +00:00 • committed by Daniel Han
commit a97135f45f
1 changed files with 13 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

22
unsloth/metrics/telemetry.py
View file

@ -81,9 +81,9 @@ def _start_telemetry_thread() -> None:
_telemetry_queue = Queue()
_telemetry_thread = threading.Thread(
target=_telemetry_worker,
daemon=True,
name="UnslothMetricsTelemetry",
target = _telemetry_worker,
daemon = True,
name = "UnslothMetricsTelemetry",
)
_telemetry_thread.start()
@ -107,7 +107,7 @@ def _telemetry_worker() -> None:
while True:
try:
item = _telemetry_queue.get(timeout=_TELEMETRY_INTERVAL)
item = _telemetry_queue.get(timeout = _TELEMETRY_INTERVAL)
except Empty:
item = "timeout"
@ -146,8 +146,12 @@ def _send_telemetry_batch() -> None:
"inference": {
"total_requests": stats["inference"].get("total_requests", 0),
"avg_e2e_latency": stats["inference"].get("avg_e2e_latency", 0.0),
"tokens_per_second": stats["inference"].get("tokens_per_second", 0.0),
"total_prompt_tokens": stats["inference"].get("total_prompt_tokens", 0),
"tokens_per_second": stats["inference"].get(
"tokens_per_second", 0.0
),
"total_prompt_tokens": stats["inference"].get(
"total_prompt_tokens", 0
),
"total_generation_tokens": stats["inference"].get(
"total_generation_tokens", 0
),
@ -175,13 +179,13 @@ def _send_to_server(payload: Dict[str, Any]) -> None:
data = json.dumps(payload).encode("utf-8")
request = urllib.request.Request(
_TELEMETRY_ENDPOINT,
data=data,
headers={
data = data,
headers = {
"Content-Type": "application/json",
"User-Agent": "Unsloth-Metrics/1.0",
},
)
urllib.request.urlopen(request, timeout=5)
urllib.request.urlopen(request, timeout = 5)
except (urllib.error.URLError, urllib.error.HTTPError, Exception):
pass