diff --git a/main.py b/main.py index 84a1a38..c9886b3 100644 --- a/main.py +++ b/main.py @@ -1,15 +1,17 @@ +import os import sys import yaml -import job +import pipelines def main(config_yaml): with open(config_yaml, 'r') as f: config = yaml.safe_load(f) + job_types = { - 'random+sobo': job.sobo_job + 'random+sobo': pipelines.sobo_pipeline } job_types[config['job']['type']](config) diff --git a/ptycho/__init__.py b/ptycho/__init__.py index 3bf318d..df223b9 100644 --- a/ptycho/__init__.py +++ b/ptycho/__init__.py @@ -1,3 +1,8 @@ from .base import PtychoEngine from .ptycho_example import ExamplePtychoEngine -from .fold_slice import FoldSlicePtychoEngine \ No newline at end of file +from .fold_slice import FoldSlicePtychoEngine + +engines = { + 'fake': ExamplePtychoEngine, + 'fold_slice': FoldSlicePtychoEngine +} \ No newline at end of file