chore: 修改类名，优化导入顺序

tests/conftest.py

@@ -11,8 +11,8 @@ from tokenizers import pre_tokenizers
 from torch.utils.data import Dataset
 
 from astrai.config.model_config import ModelConfig
-from astrai.tokenize import BpeTokenizer, BpeTrainer
 from astrai.model.transformer import Transformer
+from astrai.tokenize import BpeTokenizer, BpeTrainer
 
 
 class RandomDataset(Dataset):

tests/data/test_checkpoint.py

@@ -5,8 +5,8 @@ import torch.distributed as dist
 from torch.optim import AdamW
 from torch.optim.lr_scheduler import CosineAnnealingLR
 
-from astrai.serialization import Checkpoint
 from astrai.parallel.setup import get_rank, spawn_parallel_fn
+from astrai.serialization import Checkpoint
 
 
 def test_single_process():

tests/trainer/test_train_strategy.py

@@ -1,7 +1,7 @@
 import numpy as np
 import torch
 
-from astrai.trainer.schedule import SchedulerFactory, CosineScheduler, SGDRScheduler
+from astrai.trainer.schedule import CosineScheduler, SchedulerFactory, SGDRScheduler
 
 
 def test_schedule_factory_random_configs():