From bbdbe91871885f60d1f0ed64907470c3b8d72ed1 Mon Sep 17 00:00:00 2001
From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com>
Date: Tue, 30 Jun 2026 19:01:39 -0400
Subject: [PATCH] fix: fix dataset path calculation

---
 training/scripts/train.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/training/scripts/train.py b/training/scripts/train.py
index 4d996fd..7f473dc 100755
--- a/training/scripts/train.py
+++ b/training/scripts/train.py
@@ -61,10 +61,10 @@ def train(config_path):
     from datasets import load_dataset
     import os
     
-    # Get dataset paths relative to script location
-    script_dir = Path(__file__).parent.parent
-    train_path = str(script_dir / ".." / ".." / "training" / "data" / "train.jsonl")
-    test_path = str(script_dir / ".." / ".." / "training" / "data" / "test.jsonl")
+    # Get dataset paths - dataset is in training/data/ relative to repo root
+    repo_root = Path(__file__).parent.parent.parent
+    train_path = str(repo_root / "training" / "data" / "train.jsonl")
+    test_path = str(repo_root / "training" / "data" / "test.jsonl")
     
     dataset = load_dataset(
         "json",