fix imports

iterative · Jun 2, 2022 · 8ccc5fb · 8ccc5fb
1 parent 657067a
commit 8ccc5fb
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 4 deletions.
diff --git a/src/stages/data_load.py b/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

diff --git a/src/stages/data_split.py b/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

diff --git a/src/stages/eval.py b/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

diff --git a/src/stages/train.py b/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