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

This commit is contained in:
2026-05-12 00:34:05 +08:00
parent ba52c49edf
commit 6c96ab68c8
11 changed files with 11 additions and 13 deletions

View File

@@ -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

View File

@@ -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:

View File

@@ -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

View File

@@ -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:

View File

@@ -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"):