feat: 优化工厂模式的实现

astrai/data/__init__.py

@@ -1,7 +1,6 @@
 from astrai.data.dataset import (
     BaseDataset,
     DatasetFactory,
-    DatasetFactory,
     DPODataset,
     GRPODataset,
     MultiSegmentFetcher,

astrai/data/dataset.py

@@ -8,7 +8,7 @@ import torch
 from torch import Tensor
 from torch.utils.data import Dataset
 
-from astrai.core.factory import BaseFactory
+from astrai.factory import BaseFactory
 from astrai.data.serialization import load_h5
 
 
@@ -181,8 +181,6 @@ class DatasetFactory(BaseFactory["BaseDataset"]):
         dataset = DatasetFactory.create("custom", window_size, stride)
     """
 
-    _registry: Dict[str, type] = {}
-
     @classmethod
     def _validate_component(cls, dataset_cls: type) -> None:
         """Validate that the dataset class inherits from BaseDataset."""