Huang Qi
31fd884dfd
Drop the pre-migration PyTorch inference pipeline now that the ONNX-backed
MouseModel/ScrollModel in mouse.py and scroll.py are wired up through the
public ai_mouse API.
Deleted:
* src/ai_mouse/generator.py (legacy torch flow ODE + post-processing)
* src/ai_mouse/coord.py (legacy public coord transforms,
superseded by ai_mouse._coord)
* src/ai_mouse/_scroll_legacy.py (legacy torch scroll VAE inference)
* scripts/build_golden_*.py (one-shot capture scripts, no longer
needed once goldens are committed)
* tests/unit/test_generator.py (legacy module gone)
* tests/unit/test_scroll_generator.py (legacy module gone)
* tests/unit/test_coord.py (legacy module gone; ai_mouse._coord is
tested by test__coord.py)
* scripts/ (empty, removed)
Tools migrations:
* tools/trainer.py: import encode_trajectory from ai_mouse._coord
instead of the deleted ai_mouse.coord
* tools/server/routes_verify.py, tools/server/routes_scroll.py: route to
the public ai_mouse.generate / generate_scroll. They no longer accept
a model_dir override — the bundled ONNX is the source of truth, and a
fresh export goes through `python -m tools.export_onnx`.
* tools/eval/__main__.py: same migration; model_dir CLI arg retained as
a deprecation shim but ignored.
Final src/ai_mouse/ layout (matches plan):
__init__.py, _assets.py, _coord.py, _model_cache.py, _postprocess.py,
errors.py, mouse.py, py.typed, scroll.py, assets/
Test suite: 188 passed (was 188 before deletion; obsolete suites cleaned
out alongside the modules they covered).
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
# ai_mouse/server/routes_verify.py
|
|
"""Verification route: generate trajectories."""
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
from fastapi import APIRouter, HTTPException
|
|
from pydantic import BaseModel
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
router = APIRouter()
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Request models
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
class VerifyRequest(BaseModel):
|
|
start: list[int]
|
|
end: list[int]
|
|
n_paths: int = 5
|
|
model_dir: str | None = None
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Routes
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
@router.post("/verify")
|
|
def verify(req: VerifyRequest) -> dict:
|
|
# Uses the bundled ONNX model exposed via the public ai_mouse API.
|
|
# The legacy req.model_dir / data/models_v2 .pt path is no longer wired
|
|
# in; export a fresh .onnx via `python -m tools.export_onnx` to update.
|
|
from ai_mouse import generate
|
|
|
|
n = max(1, min(req.n_paths, 12))
|
|
start = tuple(req.start) # type: ignore[arg-type]
|
|
end = tuple(req.end) # type: ignore[arg-type]
|
|
|
|
paths = []
|
|
try:
|
|
for _ in range(n):
|
|
pts = generate(start=start, end=end)
|
|
paths.append([[x, y, t] for x, y, t in pts])
|
|
except FileNotFoundError as exc:
|
|
raise HTTPException(status_code=404, detail=str(exc)) from exc
|
|
|
|
return {"paths": paths}
|