From 7181acc5801199a1075f3b9d32bc18bb2a430938 Mon Sep 17 00:00:00 2001 From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com> Date: Tue, 30 Jun 2026 20:01:05 -0400 Subject: [PATCH] fix: save train/test.jsonl to output_dir not parent --- training/scripts/prepare_dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/training/scripts/prepare_dataset.py b/training/scripts/prepare_dataset.py index e7313aa..b10230e 100755 --- a/training/scripts/prepare_dataset.py +++ b/training/scripts/prepare_dataset.py @@ -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")