Debug entry point

1 files changed, 3 insertions, 4 deletions

diff --git a/esds/__main__.py b/esds/__main__.py
index bf05b24..4bb167d 100644
--- a/esds/__main__.py
+++ b/esds/__main__.py
@@ -1,8 +1,5 @@
 import sys, argparse, os
-import platform
-
-# Allow importlib to import file from current working directory
-sys.path.insert(0, os.getcwd())
+from .platform import YAMLPlatformFile
 
 def run(arguments):
     parser = argparse.ArgumentParser(description='Run a simulation')
@@ -10,6 +7,8 @@ def run(arguments):
     args = parser.parse_args(arguments[1:])
     if args.platform:
         simulation=YAMLPlatformFile(args.platform)
+        # Allow importlib (in simulator.run()) to import file from the platform.yaml directory
+        sys.path.insert(0, simulation.location)
         simulation.run()
     else:
         parser.print_help()