refactor: 优化参数传递，清理导入样式

tests/module/test_module.py

@@ -1,10 +1,12 @@
 import os
+
 import torch
-from astrai.trainer import *
+
 from astrai.config import *
-from astrai.model import *
 from astrai.data import *
 from astrai.inference.generator import EmbeddingEncoderCore, GeneratorCore
+from astrai.model import *
+from astrai.trainer import *
 
 
 def test_model_parameter(test_env):

tests/module/test_tie_weight.py

@@ -1,11 +1,13 @@
-import os
 import json
-import torch
-import pytest
+import os
 import tempfile
+
+import pytest
 import safetensors.torch as st
-from astrai.model.transformer import Transformer
+import torch
+
 from astrai.config.model_config import ModelConfig
+from astrai.model.transformer import Transformer
 
 
 @pytest.fixture