fix imports

Author: alex000kim

src/stages/data_load.py

@@ -1,5 +1,9 @@
-import argparse
+import sys
 from pathlib import Path
+src_path = Path(__file__).parent.parent.parent.resolve()
+sys.path.append(str(src_path))
+
+import argparse
 
 from src.data_utils import dataset_prep
 from src.load_params import load_params

src/stages/data_split.py

@@ -1,5 +1,9 @@
-import argparse
+import sys
 from pathlib import Path
+src_path = Path(__file__).parent.parent.parent.resolve()
+sys.path.append(str(src_path))
+
+import argparse
 
 from src.data_utils import create_test_dataset
 from src.load_params import load_params

src/stages/eval.py

@@ -1,6 +1,10 @@
+import sys
+from pathlib import Path
+src_path = Path(__file__).parent.parent.parent.resolve()
+sys.path.append(str(src_path))
+
 import argparse
 import json
-from pathlib import Path
 
 from src.eval_utils import get_metrics
 from src.load_params import load_params

src/stages/train.py

@@ -1,5 +1,9 @@
-import argparse
+import sys
 from pathlib import Path
+src_path = Path(__file__).parent.parent.parent.resolve()
+sys.path.append(str(src_path))
+
+import argparse
 
 from src.load_params import load_params
 from src.train_utils import train_model