style: 修改为显式导入

tests/module/test_module.py

@@ -2,11 +2,8 @@ import os
 
 import torch
 
-from astrai.config import *
-from astrai.data import *
+from astrai.config.param_config import ModelParameter
 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

@@ -36,7 +36,7 @@ def transformer_test_env():
             for file in os.listdir(test_dir):
                 os.remove(os.path.join(test_dir, file))
             os.rmdir(test_dir)
-        except:
+        except Exception:
             pass