From 1d65d240719553f2b0d281992d56c6da1251780b Mon Sep 17 00:00:00 2001
From: "xingjun.wang" <xingjun.wxj@alibaba-inc.com>
Date: Tue, 12 Dec 2023 11:52:59 +0800
Subject: [PATCH] for test

Former-commit-id: 5b979147f093e86f44c4228ab34d04bdae94f89f
---
 src/llmtuner/data/loader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/llmtuner/data/loader.py b/src/llmtuner/data/loader.py
index 70beea05..602d9a8c 100644
--- a/src/llmtuner/data/loader.py
+++ b/src/llmtuner/data/loader.py
@@ -65,7 +65,7 @@ def get_dataset(
                 example['input'] = example['input'] or ''
                 return example
 
-            dataset = dataset.ds_instance.map(map_func)
+            dataset = dataset.map(map_func)
         else:
             dataset = load_dataset(
                 path=data_path,