refactor(scroll): move trainer/models/collector to tools/scroll/

ai_mouse/scroll/__init__.py

@@ -1,6 +1,4 @@
-"""Scroll wheel event generation subsystem."""
+"""Scroll wheel event generation (inference only)."""
 from ai_mouse.scroll.generator import generate_scroll
-from ai_mouse.scroll.trainer import train_scroll
-from ai_mouse.scroll.collector import ScrollCollector
 
-__all__ = ["generate_scroll", "train_scroll", "ScrollCollector"]
+__all__ = ["generate_scroll"]

ai_mouse/scroll/generator.py

@@ -9,7 +9,7 @@ from pathlib import Path
 import numpy as np
 import torch
 
-from ai_mouse.scroll.models import ScrollCVAE
+from tools.scroll.models import ScrollCVAE
 
 logger = logging.getLogger(__name__)
 

ai_mouse/server/routes_scroll.py

@@ -85,7 +85,7 @@ def scroll_start(
     req: ScrollStartRequest,
     state: SessionState = Depends(get_state),
 ) -> dict:
-    from ai_mouse.scroll.collector import ScrollCollector
+    from tools.scroll.collector import ScrollCollector
 
     scroll_collector = ScrollCollector(
         mode=req.mode, count=req.count, viewport_height=req.viewport_height
@@ -146,7 +146,7 @@ async def _scroll_train_sse(req: ScrollTrainRequest) -> AsyncGenerator[str, None
         queue.put_nowait(msg)
 
     async def run() -> None:
-        from ai_mouse.scroll.trainer import train_scroll
+        from tools.scroll.trainer import train_scroll
 
         traces_path, models_dir = _paths()
         try:

tests/conftest.py

@@ -8,8 +8,8 @@ import numpy as np
 import pytest
 import torch
 
-from ai_mouse.scroll.models import ScrollCVAE
 from tools.models import TrajectoryFlowModel
+from tools.scroll.models import ScrollCVAE
 
 
 @pytest.fixture

tests/test_scroll_collector.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 
 import pytest
 
-from ai_mouse.scroll.collector import ScrollCollector
+from tools.scroll.collector import ScrollCollector
 
 
 class TestNextTarget:

tests/test_scroll_generator.py

@@ -8,7 +8,7 @@ import torch
 import pytest
 
 from ai_mouse.scroll.generator import generate_scroll
-from ai_mouse.scroll.models import ScrollCVAE
+from tools.scroll.models import ScrollCVAE
 
 
 @pytest.fixture

tests/test_scroll_models.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 import torch
 import pytest
 
-from ai_mouse.scroll.models import ScrollCVAE
+from tools.scroll.models import ScrollCVAE
 
 
 class TestScrollCVAEForward:

tests/test_scroll_trainer.py

@@ -8,7 +8,7 @@ from pathlib import Path
 import numpy as np
 import pytest
 
-from ai_mouse.scroll.trainer import load_scroll_data, train_scroll, _augment_scroll
+from tools.scroll.trainer import load_scroll_data, train_scroll, _augment_scroll
 
 
 def _make_synthetic_scroll_trace(mode="target"):

tools/scroll/trainer.py

@@ -20,8 +20,8 @@ import torch.nn as nn
 from torch.distributions import Normal, kl_divergence
 from torch.utils.data import DataLoader, TensorDataset
 
-from ai_mouse.scroll.models import ScrollCVAE
 from tools.config import ScrollTrainConfig
+from tools.scroll.models import ScrollCVAE
 
 logger = logging.getLogger(__name__)