o ̳iZ@srddlZddlZddlmZddlmZmZddlZddlm Z ddl m Z m Z ee jjjZGddde ZdS)N)Path)ListOptional) Subcommand)Configget_all_recipescszeZdZdZdejffdd ZdeefddZ dede e fd d Z d ed e dd fddZdejdd fddZZS)Catz2Holds all the logic for the `tune cat` subcommand. subparserscsbt|jddddtdtjd|_|jjdt dd|jj |j d |jjd d d d dS)Ncatztune catz`Pretty print a config, making it easy to know which parameters you can override with `tune run`.an examples: $ tune cat llama2/7B_full output_dir: /tmp/torchtune/llama2_7B/full tokenizer: _component_: torchtune.models.llama2.llama2_tokenizer path: /tmp/Llama-2-7b-hf/tokenizer.model max_seq_len: null ... # Pretty print the config in sorted order $ tune cat llama2/7B_full --sort # Pretty print the contents of LOCALFILE.yaml $ tune cat LOCALFILE.yaml You can now easily override a key based on your findings from `tune cat`: $ tune run full_finetune_distributed --config llama2/7B_full output_dir=./ Need to find all the "cat"-able configs? Try `tune ls`! )proghelp descriptionepilogformatter_class config_namezName of the config to print)typer )funcz--sort store_truez Print the config in sorted order)actionr ) super__init__ add_parsertextwrapdedentargparseRawTextHelpFormatter_parser add_argumentstr set_defaults_cat_cmd)selfr  __class__F/home/ubuntu/.local/lib/python3.10/site-packages/torchtune/_cli/cat.pyrs$  z Cat.__init__returncCsddtDS)NcSsg|]}|jqSr$)name).0reciper$r$r% @sz(Cat._get_all_recipes..)r)r!r$r$r%_get_all_recipes?szCat._get_all_recipes config_strcCs2tD]}|jD] }|j|kr|SqqdS)N)rconfigsr')r!r,r)configr$r$r% _get_configBs    zCat._get_configfile sort_keysNc Csz9t|d)}t|}|r&ttj|d|ddddddWdWdSWdWdS1s2wYWdStjyV}z|jd |WYd}~dSd}~ww) NrFPT)default_flow_styler1indentwidth allow_unicode)endzError parsing YAML file: )openyaml safe_loadprintdump YAMLErrorrerror)r!r0r1fdataer$r$r%_print_yaml_fileIs,  & zCat._print_yaml_fileargscCs|j}||vrtd|ddS||}|r$tdd|j}nt|}|jdvr;|j d|ddS| sK|j d |d dS| t ||jdS) aPDisplay the contents of a configuration file. Handles both predefined configurations and direct file paths, ensuring: - Input is not a recipe name - File exists - File is YAML format Args: args (argparse.Namespace): Command-line arguments containing 'config_name' attribute 'z6' is a recipe, not a config. Please use a config name.Nrecipesr->.yml.yamlzInvalid config format: 'z'. Must be YAML (.yaml/.yml)zConfig 'z ' not found.)rr+r>r/ROOT file_pathrsuffixlowerrrAexistsrErsort)r!rFr,r. config_pathr$r$r%r \s&    z Cat._cat_cmd)__name__ __module__ __qualname____doc__r_SubParsersActionrrrr+rrr/boolrE Namespacer __classcell__r$r$r"r%rs(r)rrpathlibrtypingrrr<torchtune._cli.subcommandrtorchtune._recipe_registryrr__file__parentrKrr$r$r$r%s