Ensure more general relative imports

Co-authored-by: Siddharth N <[email protected]> Co-authored-by: Brooks Paige <[email protected]>
iffsid · Dec 1, 2019 · 0de27c2 · 0de27c2
1 parent c75f099
commit 0de27c2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/report/analyse_cub.py b/src/report/analyse_cub.py
@@ -10,7 +10,8 @@
 import inspect
 currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
 parentdir = os.path.dirname(currentdir)
-sys.path.insert(0, parentdir)
+sys.path.insert(0, parentdir) # for system user
+os.chdir(parentdir) # for pycharm user
 
 import models
 from utils import Logger, Timer, unpack_data

diff --git a/src/report/analyse_ms.py b/src/report/analyse_ms.py
@@ -13,7 +13,8 @@
 import inspect
 currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
 parentdir = os.path.dirname(currentdir)
-sys.path.insert(0, parentdir)
+sys.path.insert(0, parentdir) # for system user
+os.chdir(parentdir) # for pycharm user
 
 import models
 from helper import Latent_Classifier, SVHN_Classifier, MNIST_Classifier