PiddlZddlZddlZddlmZddlZddlmZddlm Z eej j j Z GddeZ dS)N)Path) Subcommand)get_all_recipesc@eZdZdZfdZddZdejfdZxZ S) Copyz1Holds all the logic for the `tune cp` subcommand.c *t|dddddtjdt j|_||j |j dS)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__s e/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchtune/_cli/cp.pyrz Copy.__init__s !,, ADK?   %9%-   (  !!t|!44444returnNc|jdtd|jdtd|jdddd d |jd dd 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)rs rrzCopy._add_arguments.s !! [ "   !! / "   !!  D "    !! P "     r argsc|j}d}tD][}|j|jkrtdz |jz }d}n4|jD]+}|j|jkrtdz dz |jz }d}n,\|#|jd|jd|jdkr |j |kr| |} |j r;| r'td |d dS|jr|jd d t%j||}td |dS#t($r"|jd|dYdSwxYw)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_suffixconfigoutputs rrz Copy._cp_cmdIs , &''  F{di''Y&)99 %  .  ;$)++*Y69IIC$+ME, ; L  ddiddd     r ! !k&8M&I&I%11-@@K  2;#5#5#7#7 2Xk.B.B.D.DXXX$J&,,TD,IIIS+6600011111     L  M MMM       sAEA E(E?>E?)r!N) __name__ __module__ __qualname____doc__rrr Namespacer __classcell__)rs@rrrsp;;555552    6+H.++++++++r r)rr?rpathlibr torchtunetorchtune._cli.subcommandrtorchtune._recipe_registryr__file__r=r3rr rrSs  000000666666 tI &-bbbbb:bbbbbr