refactor: move trainer/models/utils/config to tools/

2026-05-12 00:30:55 +08:00
parent c89025047c
commit ba52c49edf
13 changed files with 28 additions and 28 deletions
--- a/tests/test_generator.py
+++ b/tests/test_generator.py
@@ -9,7 +9,7 @@ import pytest
 import torch

 from ai_mouse.generator import generate
-from ai_mouse.models import TrajectoryFlowModel
+from tools.models import TrajectoryFlowModel


@pytest.fixture