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_models.py
+++ b/tests/test_models.py
@@ -4,7 +4,7 @@ from __future__ import annotations
 import torch
 import pytest

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


 class TestTrajectoryFlowModel: