o 㥵i! @sddlmZddlmZddlZddlZddlZddlmZddl m Z ddl m Z m Z mZddlmZddlmZe dd e d eed ededdf ddZe dd e deddfddZdS))Path)SequenceN) HydraConfig)rank_zero_only) DictConfig OmegaConf open_dict)Prompt)loggerdatamodel callbacksr trainerpathsextrasFcfg print_orderresolve save_to_filereturnc Cs"d}tjjd||d}g}|D]}||vr||ntd|dd|dq|D] }||vr7||q,|D])}|j|||d}||} t| trUt j | |d} nt | } |tj | d q:t||rtt|jjd d } tj|| d Wd d S1swYd Sd S)aPrints content of DictConfig using Rich library and its tree structure. Args: cfg (DictConfig): Configuration composed by Hydra. print_order (Sequence[str], optional): Determines in what order config components are printed. resolve (bool, optional): Whether to resolve reference fields of DictConfig. save_to_file (bool, optional): Whether to export config to the hydra output folder. dimCONFIG)style guide_stylezField 'z' not found in config. z Skipping 'z' config printing...)ryamlzconfig_tree.logwfileN)richtreeTreeappendlogwarningadd isinstancerrto_yamlstrsyntaxSyntaxprintopenrr output_dir) rrrrrr queuefieldbranch config_groupbranch_contentrr3P/home/ubuntu/.local/lib/python3.10/site-packages/fish_speech/utils/rich_utils.pyprint_config_trees:      "r5cCs|dsGdtjjjvrtdtdtj ddd}dd | d D}t | ||_ Wd n1s9wYt d |j |rntt|jjd d}tj|j |dWd d S1sgwYd Sd S)zOPrompts user to input tags from command line if no tags are provided in config.tagsidz)Specify tags before launching a multirun!z;No tags provided in config. Prompting user to input tags...z$Enter a list of comma separated tagsdev)defaultcSsg|] }|dkr|qS))strip).0tr3r3r4 [sz enforce_tags..,NzTags: ztags.logrr)getrrhydrajob ValueErrorr#r$r asksplitrr6infor,rrr-rr+)rrr6rr3r3r4 enforce_tagsQs   "rG)r FF)F)pathlibrtypingrr rich.syntax rich.treehydra.core.hydra_configrlightning.pytorch.utilitiesr omegaconfrrr rich.promptr fish_speech.utilsr r#r(boolr5rGr3r3r3r4s6         A