@@ -87,7 +87,7 @@ class ModelTrainer:
data.append(json.loads(line))
# 将列表转换为 HuggingFace Dataset 对象
- data = Dataset.from_list(data)
+ train_dataset = Dataset.from_list(data)
return train_dataset