mirror of
https://github.com/c-sooyoung/bo-ptycho.git
synced 2026-09-17 21:29:07 +09:00
27 lines
651 B
Python
27 lines
651 B
Python
import os
|
|
import sys
|
|
import yaml
|
|
import shutil
|
|
|
|
import pipelines
|
|
|
|
def main(config_yaml):
|
|
with open(config_yaml, 'r') as f:
|
|
config = yaml.safe_load(f)
|
|
|
|
result_dir = config["io"]["result_dir"]
|
|
if os.path.exists(result_dir):
|
|
shutil.rmtree(result_dir)
|
|
os.makedirs(result_dir, exist_ok=True)
|
|
shutil.copy(config_yaml, os.path.join(result_dir, os.path.basename(config_yaml)))
|
|
|
|
pipelines.job_types[config['job']['type']](config)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) != 2:
|
|
print("Usage: python bo-ptycho.py <config_yaml>")
|
|
sys.exit(1)
|
|
config_yaml = sys.argv[1]
|
|
main(config_yaml)
|