yellowdog/ai_mouse
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: move CLI dispatcher to tools/__main__.py

Browse Source
This commit is contained in:
Huang Qi
2026-05-12 00:42:18 +08:00
parent 3507fdc202
commit 31ad47dc0b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ai_mouse/__main__.py → tools/__main__.py
View File

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