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

import os
import sys

SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
PROJECT_ROOT = os.path.dirname(SCRIPT_DIR)
sys.path.append(PROJECT_ROOT)

from uf_data.load import UFConfigLoader
from uf_data.pipeline import UFAnalysisPipeline


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

    # 1. 加载配置文件
    config_path = os.path.join(PROJECT_ROOT, "config", "uf_analyze_config.yaml")
    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__":
    main()