bvandeusen
bc6d43d3f2
Hygiene follow-up to the Camie retirement (#1189) — these were left inert to bound that change; nothing reads them now. Migration 0068 drops: - ml_settings: tagger_store_floor, tagger_model_version, suggestion_threshold_ character/general (already dead pre-retirement — scoring uses per-head thresholds), video_min_tag_frames (only the deleted video-prediction aggregator used it). - image_record: tagger_model_version (no writer), centroid_scores (dead JSON cache, no reader). Also: ml_admin _EDITABLE/GET/_validate pruned (dropped the store-floor invariant + video_min_tag_frames check); MLThresholdSliders trimmed to a video-embedding card (interval + max frames only); importer no longer resets the dropped cols; download_models drops the Camie fetch; stale CASCADE comments in cleanup_service no longer name the removed tables. Tests updated. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01Ttrj5P7upUTueSfoJcxEqa
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
"""download_models tests. No network in CI; we test the 'already present →
|
|
skip' short-circuit by faking the expected files, and that main() wires the
|
|
SigLIP fetch. (Camie download retired with the tagger, #1199.)
|
|
"""
|
|
|
|
from unittest.mock import patch
|
|
|
|
from backend.app.scripts import download_models as dm
|
|
|
|
|
|
def test_ensure_siglip_skips_when_present(tmp_path, monkeypatch):
|
|
monkeypatch.setattr(dm, "MODEL_ROOT", tmp_path)
|
|
sig = tmp_path / "siglip"
|
|
sig.mkdir(parents=True)
|
|
(sig / "config.json").write_text("{}")
|
|
(sig / "model.safetensors").write_bytes(b"x")
|
|
with patch.object(dm, "_snapshot") as snap:
|
|
dm.ensure_siglip()
|
|
snap.assert_not_called()
|
|
|
|
|
|
def test_ensure_siglip_downloads_when_missing(tmp_path, monkeypatch):
|
|
monkeypatch.setattr(dm, "MODEL_ROOT", tmp_path)
|
|
with patch.object(dm, "_snapshot") as snap:
|
|
dm.ensure_siglip()
|
|
snap.assert_called_once()
|
|
|
|
|
|
def test_main_fetches_siglip(monkeypatch):
|
|
calls = []
|
|
monkeypatch.setattr(dm, "ensure_siglip", lambda: calls.append("siglip"))
|
|
assert dm.main() == 0
|
|
assert calls == ["siglip"]
|