fix(nyz): fix import bugs

opendilab · Jan 27, 2025 · 1295987 · 1295987
1 parent 702f244
commit 1295987
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/ding/utils/data/__init__.py b/ding/utils/data/__init__.py
@@ -2,3 +2,5 @@
 from .dataloader import AsyncDataLoader
 from .dataset import NaiveRLDataset, D4RLDataset, HDF5Dataset, BCODataset, \
     create_dataset, hdf5_save, offline_data_save_type
+from .rlhf_online_dataset import OnlineRLDataset
+from .rlhf_offline_dataset import OfflineRLDataset
diff --git a/ding/utils/data/tests/test_rlhf_offline_dataset.py b/ding/utils/data/tests/test_rlhf_offline_dataset.py
@@ -1,6 +1,6 @@
 import pytest
 from datasets import load_dataset, concatenate_datasets
-from rl.data.offlinerl_dataset import OfflineRLDataset
+from ding.utils.data import OfflineRLDataset
 from transformers import AutoTokenizer
 
 

diff --git a/ding/utils/data/tests/test_rlhf_online_dataset.py b/ding/utils/data/tests/test_rlhf_online_dataset.py
@@ -1,7 +1,7 @@
 import pytest
 from datasets import load_dataset
 from transformers import AutoTokenizer
-from rl.data.onlinerl_dataset import OnlineRLDataset
+from ding.utils.data import OnlineRLDataset
 
 
 @pytest.fixture