diff --git a/mains/_init_paths.py b/mains/_init_paths.py new file mode 100644 index 0000000..026987d --- /dev/null +++ b/mains/_init_paths.py @@ -0,0 +1,11 @@ +import os +import sys + +def add_path(path): + if path not in sys.path: + sys.path.insert(0, path) + +this_dir = os.getcwd() + +root_path = os.path.join(this_dir, '..', '') +add_path(root_path) diff --git a/mains/example.py b/mains/example.py index 6f8670b..f5a20b0 100644 --- a/mains/example.py +++ b/mains/example.py @@ -1,5 +1,6 @@ import tensorflow as tf +import _init_paths from data_loader.data_generator import DataGenerator from models.example_model import ExampleModel from trainers.example_trainer import ExampleTrainer