add mypy check

Author: binary-husky

.pre-commit-config.yaml

@@ -22,3 +22,21 @@ repos:
           --remove-all-unused-imports,
           --ignore-init-module-imports
         ]
+
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v1.7.0
+    hooks:
+      - id: mypy
+        args: [
+          --ignore-missing-imports,
+          --disable-error-code=var-annotated,
+          --disable-error-code=union-attr,
+          --disable-error-code=no-redef,
+          --disable-error-code=assignment,
+          --disable-error-code=has-type,
+          --disable-error-code=attr-defined,
+          --disable-error-code=import-untyped,
+          --disable-error-code=truthy-function,
+          --follow-imports=skip,
+          --explicit-package-bases,
+        ]

ajet/backbone/trainer_trinity.py

@@ -1,12 +1,12 @@
-import asyncio
 import os
-from typing import Dict, List, Literal, Optional, cast
-
+import asyncio
 import datasets
 import openai
 import swanlab
+
 from loguru import logger
 from transformers import AutoTokenizer
+from typing import Dict, List, Literal, Optional, cast
 from trinity.buffer.reader import READER
 from trinity.buffer.reader.file_reader import TaskFileReader, _HFBatchReader
 from trinity.buffer.schema import FORMATTER
@@ -19,9 +19,7 @@
 from trinity.utils.monitor import MONITOR, Monitor
 
 from ajet.backbone.warm_up import warm_up_process
-from ajet.context_tracker.multiagent_tracking import (
-    MultiAgentContextTracker,
-)
+from ajet.context_tracker.multiagent_tracking import MultiAgentContextTracker
 from ajet.schema.trajectory import Sample
 from ajet.task_reader import dict_to_ajet_task
 from ajet.task_rollout.native_parallel_worker import DynamicRolloutManager