Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

print the config first

Browse Source
This commit is contained in:
goulustis
2026-03-03 14:40:03 +08:00
parent 7294e07df7
commit 62a00b4a5b
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/py_scripts/misc_tasks.py
View File

@@ -7,6 +7,7 @@ import os
from tqdm import tqdm from tqdm import tqdm
import yaml import yaml
import tyro import tyro
from loguru import logger
def gen_arch_imgs(save_dir="frontend/assets/images/graph_arch"): def gen_arch_imgs(save_dir="frontend/assets/images/graph_arch"):
@@ -17,9 +18,11 @@ def gen_arch_imgs(save_dir="frontend/assets/images/graph_arch"):
img = graph.show_graph(ret_img=True) img = graph.show_graph(ret_img=True)
img.save(osp.join(save_dir, f"arch_{conf.__class__.__name__}.png")) img.save(osp.join(save_dir, f"arch_{conf.__class__.__name__}.png"))
def make_save_conf(pipeline_config:PipelineConfig, save_path:str):
def make_save_conf(pipeline:PipelineConfig, save_path:str):
os.makedirs(osp.dirname(save_path), exist_ok=True) os.makedirs(osp.dirname(save_path), exist_ok=True)
pipeline_config.save_config(save_path) logger.info(pipeline)
pipeline.save_config(save_path)
if __name__ == "__main__": if __name__ == "__main__":
# gen_arch_imgs() # gen_arch_imgs()