From 31ad47dc0b4a7e58e952fa5e704faed4ca58475d Mon Sep 17 00:00:00 2001 From: Huang Qi Date: Tue, 12 May 2026 00:42:18 +0800 Subject: [PATCH] refactor: move CLI dispatcher to tools/__main__.py --- {ai_mouse => tools}/__main__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) rename {ai_mouse => tools}/__main__.py (95%) diff --git a/ai_mouse/__main__.py b/tools/__main__.py similarity index 95% rename from ai_mouse/__main__.py rename to tools/__main__.py index a60eb5d..030718e 100644 --- a/ai_mouse/__main__.py +++ b/tools/__main__.py @@ -12,7 +12,7 @@ from pathlib import Path def _train_main(args: argparse.Namespace) -> int: - from ai_mouse.trainer import train + from tools.trainer import train logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s %(message)s") train( @@ -28,7 +28,7 @@ def _train_main(args: argparse.Namespace) -> int: def _eval_main(args: argparse.Namespace) -> int: - from ai_mouse.eval.__main__ import main as eval_main + from tools.eval.__main__ import main as eval_main # Reconstruct argv for the sub-CLI argv = [ "--model-dir", args.model_dir, @@ -43,7 +43,7 @@ def _eval_main(args: argparse.Namespace) -> int: def _balabit_main(args: argparse.Namespace) -> int: - from ai_mouse.data_adapters.balabit import main as bal_main + from tools.data_adapters.balabit import main as bal_main argv = [ "--input", str(args.input), "--output", str(args.output),