add add_same_content_to_new_column reference

modelscope · Jan 23, 2025 · 530efa8 · 530efa8
1 parent dd95df0
commit 530efa8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/data_juicer/core/data/__init__.py b/data_juicer/core/data/__init__.py
@@ -1,6 +1,9 @@
-from .dj_dataset import DJDataset, NestedDataset, wrap_func_with_nested_access
+from .dj_dataset import (DJDataset, NestedDataset,
+                         add_same_content_to_new_column,
+                         wrap_func_with_nested_access)
 from .ray_dataset import RayDataset
 
 __all__ = [
-    'DJDataset', 'NestedDataset', 'RayDataset', 'wrap_func_with_nested_access'
+    'DJDataset', 'NestedDataset', 'RayDataset', 'wrap_func_with_nested_access',
+    'add_same_content_to_new_column'
 ]