unsloth/studio/backend/utils/downsample.py
Daniel Han 3ce187da02
Formatting: ruff line-length 100, kwarg-spacing passes, drop blank after short local imports (#6079)
Raise ruff line-length to 100 and extend the local pre-commit format pipeline (def-signature magic-comma normalization, short multi-line assert collapse, kwarg '=' spacing, blank-line-after-short-import removal, adjacent string-literal / f-string+plain merge, redundant-pass pruning). Every transform re-checks the file AST and is dropped if it would differ; the whole-repo reformat is verified AST-identical per file and idempotent.
2026-06-08 04:24:13 -07:00

16 lines
638 B
Python

# SPDX-License-Identifier: AGPL-3.0-only
# Copyright 2026-present the Unsloth AI Inc. team. All rights reserved. See /studio/LICENSE.AGPL-3.0
"""Generic numeric downsampling utility."""
def downsample(values: list[float], target_count: int) -> list[float]:
"""Reduce a list to target_count points via evenly-spaced index sampling."""
if len(values) <= target_count:
return list(values)
if target_count <= 0:
return []
if target_count == 1:
return [values[-1]]
indices = [round(i * (len(values) - 1) / (target_count - 1)) for i in range(target_count)]
return [values[i] for i in indices]