From 383125edc97d5fcb998d5ceb8c0679bb2315e692 Mon Sep 17 00:00:00 2001
From: Colin <colin>
Date: Tue, 18 Feb 2025 14:21:15 +0800
Subject: [PATCH] Refine dataset org.

---
 wit/{ => dataset}/dataset.py         | 4 ++--
 wit/{ => dataset}/meaning_dataset.py | 0
 wit/{ => dataset}/special_dataset.py | 0
 wit/train.py                         | 2 +-
 4 files changed, 3 insertions(+), 3 deletions(-)
 rename wit/{ => dataset}/dataset.py (92%)
 rename wit/{ => dataset}/meaning_dataset.py (100%)
 rename wit/{ => dataset}/special_dataset.py (100%)

diff --git a/wit/dataset.py b/wit/dataset/dataset.py
similarity index 92%
rename from wit/dataset.py
rename to wit/dataset/dataset.py
index a9effde..3979b06 100644
--- a/wit/dataset.py
+++ b/wit/dataset/dataset.py
@@ -1,5 +1,5 @@
-from meaning_dataset import MeaningDataset, BatchGroupMeaningDataloader
-from special_dataset import SpecialDataset
+from dataset.meaning_dataset import MeaningDataset, BatchGroupMeaningDataloader
+from dataset.special_dataset import SpecialDataset
 from torch.utils.data import random_split, DataLoader
 
 
diff --git a/wit/meaning_dataset.py b/wit/dataset/meaning_dataset.py
similarity index 100%
rename from wit/meaning_dataset.py
rename to wit/dataset/meaning_dataset.py
diff --git a/wit/special_dataset.py b/wit/dataset/special_dataset.py
similarity index 100%
rename from wit/special_dataset.py
rename to wit/dataset/special_dataset.py
diff --git a/wit/train.py b/wit/train.py
index 88a157d..f8b6539 100644
--- a/wit/train.py
+++ b/wit/train.py
@@ -6,7 +6,7 @@ from wit.model.tokenization_qwen import QWenTokenizer
 from logger import MLFLogger
 
 import configuration
-import dataset as ds
+import dataset.dataset as ds
 
 if __name__ == "__main__":