diff options
Diffstat (limited to 'esds/__main__.py')
| -rw-r--r-- | esds/__main__.py | 7 |
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() |