DualFlow/models/uf-rl/uf_data_process/run_ufdata_pipeline.py

import os
import sys
from pathlib import Path

THIS_FILE = Path(__file__).resolve()
UF_RL_ROOT = THIS_FILE.parents[1]

from load import UFConfigLoader
from pipeline import UFAnalysisPipeline


def main():
    print("=====================================")
    print("    UF Multi-Unit Analysis Pipeline")
    print("=====================================")

    # 1. 加载配置文件
    cfg = UFConfigLoader(CONFIG_PATH)

    # 2. 创建 pipeline
    pipeline = UFAnalysisPipeline(cfg)

    # 3. 执行 pipeline
    print(">>> Running pipeline ...")
    results = pipeline.run()

    print("=========================================")
    print("Pipeline finished. Results saved.")
    print("=========================================")


if __name__ == "__main__":
    CONFIG_PATH = UF_RL_ROOT / "yancheng" / "uf_analyze_config.yaml"
    main()