o ̳i@sfddlZddlZddlZddlmZddlZddlmZddlm Z eej j j Z GdddeZ dS)N)Path) Subcommand)get_all_recipescs:eZdZdZfddZd ddZdejfd d ZZ S) Copyz1Holds all the logic for the `tune cp` subcommand.c sHt|jdddddtdtjd|_||jj |j ddS)Ncpztune cpz-tune cp destination [OPTIONS]z1Copy a built-in recipe or config to a local path.a examples: $ tune cp lora_finetune_distributed . Copied file to ./lora_finetune_distributed.py $ tune cp llama2/7B_full ./new_dir/my_custom_lora.yaml --make-parents Copyied file to ./new_dir/my_custom_lora.yaml Need to see all possible recipes/configs to copy? Try running `tune ls`. )progusagehelp descriptionepilogformatter_class)func) super__init__ add_parsertextwrapdedentargparseRawTextHelpFormatter_parser_add_arguments set_defaults_cp_cmd)self subparsers __class__E/home/ubuntu/.local/lib/python3.10/site-packages/torchtune/_cli/cp.pyrs  z Copy.__init__returnNcCsR|jjdtdd|jjdtdd|jjdddd d d |jjd dd d d dS)zAdd arguments to the parser.filezHRecipe/config to copy. For a list of all possible options, run `tune ls`)typer destinationzLocation to copy the file toz-nz --no-clobber store_truez1Do not overwrite destination if it already existsF)actionr defaultz--make-parentszCreate parent directories for destination if they do not exist. If not set to True, will error if parent directories do not existN)r add_argumentstrr)rrrrr.s. zCopy._add_argumentsargscCs.|j}d}tD],}|j|jkrtd|j}d}n|jD]}|j|jkr3tdd|j}d}nqq|durD|jd|jd|jdkrS|j |krS| |}z/|j rh| rht d |d WdS|jrs|jjd d d t||}t d |WdSty|jd|dYdSw)z*Copy a recipe or config to a new location.Nrecipesz.pyconfigsz.yamlzInvalid file name: z3. Try `tune ls` to see all available files to copy.zFile already exists at z, not overwriting.T)parentsexist_okzCopied file to zCannot create regular file: 'z'. No such file or directory. If the specified destination's parent directory does not exist and you would like to create it on-the-fly, use the --make-parents flag.)r"rnamer ROOT file_pathr*rerrorsuffix with_suffix no_clobberexistsprintabsolute make_parentsparentmkdirshutilcopyFileNotFoundError)rr(r"srcrecipe proper_suffixconfigoutputrrrrIsD           z Copy._cp_cmd)rN) __name__ __module__ __qualname____doc__rrr Namespacer __classcell__rrrrrs   r)rr;rpathlibr torchtunetorchtune._cli.subcommandrtorchtune._recipe_registryr__file__r9r/rrrrrs