From 79a376d1db4eb02a9c584daee75bcdfc2d7ed333 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: 33d9082320098f994bfa0c6353459afcb93165b7
---
 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,