Update data_args.py

This commit is contained in:
hiyouga
2023-07-28 17:42:41 +08:00
parent 5ee87138e4
commit 513e1f1ec9

View File

@@ -116,4 +116,4 @@ class DataArguments:
dataset_attr.response_column = dataset_info[name]["columns"].get("response", None) dataset_attr.response_column = dataset_info[name]["columns"].get("response", None)
dataset_attr.history_column = dataset_info[name]["columns"].get("history", None) dataset_attr.history_column = dataset_info[name]["columns"].get("history", None)
self.dataset_list.append(dataset_attr) self.dataset_list.append(dataset_attr)