refactor(scroll): move trainer/models/collector to tools/scroll/
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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:
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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:
|
||||
|
||||
@@ -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"):
|
||||
|
||||
Reference in New Issue
Block a user