style: 修改为显式导入

2026-04-04 16:02:49 +08:00
parent 3346c75584
commit b531232a9b
12 changed files with 38 additions and 34 deletions
@@ -1,9 +1,7 @@
 import numpy as np
 import torch

-from astrai.config import *
-from astrai.data.dataset import *
-from astrai.trainer.schedule import *
+from astrai.trainer.schedule import SchedulerFactory, CosineScheduler, SGDRScheduler


 def test_schedule_factory_random_configs():