sys path

2026-05-23 13:57:26 +02:00
parent 882c2b53b6
commit d4a418fbd9
1 changed files with 5 additions and 0 deletions
@@ -1,11 +1,16 @@
 import argparse
 import itertools
 import os
+import sys
 import tempfile

 import yaml
 import submitit

+PROJECT_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+if PROJECT_ROOT not in sys.path:
+    sys.path.insert(0, PROJECT_ROOT)
+
 from src.train_supervised import main as app_main