o ̳i @s<ddlZddlZddlmZddlmZGdddeZdS)N) Subcommand)get_all_recipescs@eZdZdZdZdejffdd Zdejddfd d Z Z S) Listz1Holds all the logic for the `tune ls` subcommand.z<> subparserscs>t|jddddtdtjd|_|jj|j ddS)Nlsztune lsz%List all built-in recipes and configsa examples: $ tune ls RECIPE CONFIG full_finetune_single_device llama2/7B_full_single_device full_finetune_distributed llama2/7B_full llama2/13B_full ... To run one of these recipes: $ tune run full_finetune_single_device --config full_finetune_single_device )proghelp descriptionepilogformatter_class)func) super__init__ add_parsertextwrapdedentargparseRawTextHelpFormatter_parser set_defaults_ls_cmd)selfr __class__E/home/ubuntu/.local/lib/python3.10/site-packages/torchtune/_cli/ls.pyrs z List.__init__argsreturnNcCsddddd}t|tD]8}|j}t|jdkr*|dd|jd}t|t|jD]\}}|dkr9d}|dd|jd}t|q/qdS)z'List all available recipes and configs.RECIPEz<40 CONFIGrN)printrnamelenconfigs NULL_VALUE enumerate)rrheaderrecipe recipe_strrowiconfigrrrr-s  z List._ls_cmd) __name__ __module__ __qualname____doc__r&r_SubParsersActionr Namespacer __classcell__rrrrrs r)rrtorchtune._cli.subcommandrtorchtune._recipe_registryrrrrrrs