cmedina/agenx-lora-training
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: save train/test.jsonl to output_dir not parent

Browse Source
This commit is contained in:
Christian Medina
2026-06-30 20:01:05 -04:00
parent d5ba471b98
commit 7181acc580
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
training/scripts/prepare_dataset.py
View File

@@ -62,11 +62,11 @@ def prepare_dataset(input_file, output_file, test_size=0.05):
test_data = formatted[split_point:]
# Save
with open(output_file.parent / "train.jsonl", "w") as f:
with open(output_dir / "train.jsonl", "w") as f:
for item in train_data:
f.write(json.dumps(item) + "\n")
with open(output_file.parent / "test.jsonl", "w") as f:
with open(output_dir / "test.jsonl", "w") as f:
for item in test_data:
f.write(json.dumps(item) + "\n")