chore: 将data 模块命名为dataset

2026-04-04 16:16:27 +08:00
parent bd9741dc5f
commit 9c31d78a22
8 changed files with 7 additions and 7 deletions
@@ -0,0 +1,19 @@
+from astrai.dataset.dataset import (
+    BaseDataset,
+    DatasetFactory,
+    BaseSegmentFetcher,
+    MultiSegmentFetcher,
+)
+from astrai.dataset.sampler import ResumableDistributedSampler
+
+__all__ = [
+    # Base classes
+    "BaseDataset",
+    # Factory
+    "DatasetFactory",
+    # Fetchers
+    "BaseSegmentFetcher",
+    "MultiSegmentFetcher",
+    # Sampler
+    "ResumableDistributedSampler",
+]