mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-08-05 02:45:28 +00:00
cf4e240ad1
Some checks failed
CI / Focused test guidance (report-only) (push) Has been cancelled
CI / Python syntax (compileall) (push) Has been cancelled
CI / JS syntax (node --check) (push) Has been cancelled
CI / Python tests (pytest) (push) Has been cancelled
CodeQL / Analyze (actions) (push) Has been cancelled
CodeQL / Analyze (javascript-typescript) (push) Has been cancelled
CodeQL / Analyze (python) (push) Has been cancelled
Container scan / hadolint (Dockerfile lint) (push) Has been cancelled
Container scan (Trivy) / Trivy (image scan, advisory) (push) Has been cancelled
Container scan (Trivy) / Trivy (image scan + SARIF upload) (push) Has been cancelled
Dependency review / dependency-review (PR gate) (push) Has been cancelled
Dependency review / pip-audit (advisory) (push) Has been cancelled
ci / docker publish / build (amd64) (push) Has been cancelled
ci / docker publish / build (arm64) (push) Has been cancelled
Secret scan / gitleaks (push) Has been cancelled
Workflow security / actionlint (push) Has been cancelled
Workflow security / zizmor (Actions SAST) (push) Has been cancelled
ci / docker publish / merge manifest + tag (push) Has been cancelled
31 lines
1.3 KiB
Python
31 lines
1.3 KiB
Python
"""Harden hwfit model-catalog parsing against non-string field values.
|
|
|
|
`params_b` and `is_prequantized` read free-form fields straight off the HF
|
|
catalog JSON. `parameter_count` is normally a string like "7B" and
|
|
`quantization` a string like "FP8", but a catalog row can carry a non-string
|
|
(e.g. an integer parameter_count, or a null/number quantization). The code
|
|
called `pc.strip()` / `q.startswith(...)` directly, so one such row raised
|
|
AttributeError and aborted the whole ranking pass (params_b/is_prequantized
|
|
run for every model). Non-strings are now treated as unknown.
|
|
"""
|
|
from services.hwfit.models import params_b, is_prequantized
|
|
|
|
|
|
def test_params_b_nonstring_count_does_not_raise():
|
|
assert params_b({"parameter_count": 7}) == 0.0
|
|
assert params_b({"parameter_count": ["7B"]}) == 0.0
|
|
|
|
|
|
def test_params_b_valid_count_still_parses():
|
|
assert params_b({"parameter_count": "7B"}) == 7.0
|
|
assert params_b({"parameters_raw": 7_000_000_000}) == 7.0
|
|
|
|
|
|
def test_is_prequantized_nonstring_quantization_does_not_raise():
|
|
assert is_prequantized({"quantization": 8}) is False
|
|
assert is_prequantized({"name": "plain-model", "quantization": 123}) is False
|
|
|
|
|
|
def test_is_prequantized_still_detects_real_markers():
|
|
assert is_prequantized({"name": "some-model-awq"}) is True
|
|
assert is_prequantized({"quantization": "FP8-Mixed"}) is True
|