From c8df8d56116585f646c094b55d0ec11ce4b1dcea Mon Sep 17 00:00:00 2001
From: Sooyoung Cheong <64125280+c-sooyoung@users.noreply.github.com>
Date: Wed, 5 Aug 2026 14:27:19 +0900
Subject: [PATCH] fixes

---
 main.py            | 6 ++++--
 ptycho/__init__.py | 7 ++++++-
 2 files changed, 10 insertions(+), 3 deletions(-)

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