feat: 优化工厂模式的实现

astrai/trainer/schedule.py

@@ -6,7 +6,7 @@ from typing import Any, Dict, List, Type
 
 from torch.optim.lr_scheduler import LRScheduler
 
-from astrai.core.factory import BaseFactory
+from astrai.factory import BaseFactory
 
 
 class BaseScheduler(LRScheduler, ABC):
@@ -41,8 +41,6 @@ class SchedulerFactory(BaseFactory["BaseScheduler"]):
         scheduler = SchedulerFactory.create("custom", optimizer, **kwargs)
     """
 
-    _registry: Dict[str, Type[BaseScheduler]] = {}
-
     @classmethod
     def _validate_component(cls, scheduler_cls: Type[BaseScheduler]) -> None:
         """Validate that the scheduler class inherits from BaseScheduler."""