Jayant-Kernel commited on
Commit
91f4c95
·
1 Parent(s): 3fdbb3c

fix: use dynamic path for level2.jsonl

Browse files
Files changed (1) hide show
  1. train.py +2 -1
train.py CHANGED
@@ -201,7 +201,8 @@ print("Level 1 checkpoint saved locally")
201
 
202
  # Load Level 2 dataset
203
  import pathlib as _pl2
204
- data_path_l2 = _pl2.Path("/home/trainer/.local/lib/python3.10/site-packages/deceit_env/data/level2.jsonl")
 
205
  questions_l2 = []
206
  with open(data_path_l2) as f:
207
  for line in f:
 
201
 
202
  # Load Level 2 dataset
203
  import pathlib as _pl2
204
+ import deceit_env as _de2
205
+ data_path_l2 = _pl2.Path(_de2.__file__).parent / "data" / "level2.jsonl"
206
  questions_l2 = []
207
  with open(data_path_l2) as f:
208
  for line in f: