use main

Begunner · Begunner · commit b03bc7e93231 · 2025-12-30T15:33:32.000+08:00
diff --git a/examples/data_preprocess/geo3k.py b/examples/data_preprocess/geo3k.py
@@ -37,9 +37,7 @@
     data_source = "hiyouga/geometry3k"
 
     if local_dataset_path is not None:
-        dataset = datasets.load_dataset(
-            local_dataset_path,
-        )
+        dataset = datasets.load_dataset(local_dataset_path, "main")
     else:
         dataset = datasets.load_dataset(
             data_source,
