fix: fixed sys.path issues on running generate_data.py

training/generate_data.py

@@ -6,9 +6,13 @@ import logging
 import os
 import random
 import statistics
+import sys
 from collections import Counter, defaultdict
 from pathlib import Path
 
+# Repo root on path when run as `python training/generate_data.py` (script dir is training/)
+sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
+
 from dotenv import load_dotenv
 
 from agent.runner import EpisodeResult, run_episode