Update data_args.py

Former-commit-id: 513e1f1ec952dc5901d90dd34e32440230bca3f0
This commit is contained in:
hiyouga 2023-07-28 17:42:41 +08:00
parent f65f0745cc
commit 2a664783c3

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)