o 0i6@sbddlZGdddejZGdddeZGdddeZGdd d ejZGd d d ejZdS) Ncs*eZdZdZfddZdddZZS) _StoreActionzS Custom action that allows for `-` or `_` to be passed in for an argument. csXtj|i|g}|jD]}||d|ddvr&||ddq||_dS)N_-)super__init__option_stringsappendreplace)selfargskwargsnew_option_strings option_string __class__[/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/accelerate/commands/utils.pyrs   z_StoreAction.__init__NcCs2t||j|t|dst|_|j|jdS)N nondefault)setattrdesthasattrsetraddr parser namespacevaluesrrrr__call__!s z_StoreAction.__call__N__name__ __module__ __qualname____doc__rr __classcell__rrrrrs  rcs0eZdZdZdfdd Zd fdd ZZS) _StoreConstActionzR Same as `argparse._StoreConstAction` but uses the custom `_StoreAction`. NFc stj||d||||ddS)Nr)rrnargsconstdefaultrequiredhelprr)r rrr(r)r*r+rrrr-s z_StoreConstAction.__init__cst|||j|dSr)rrr(rrrrr8sz_StoreConstAction.__call__NFNrr rrrrr&(s r&cs(eZdZdZ   dfdd ZZS)_StoreTrueActionzV Same as `argparse._StoreTrueAction` but uses the custom `_StoreConstAction`. NFcstj||d|||ddS)NT)rrr(r)r*r+r,)r rrr)r*r+rrrrAs  z_StoreTrueAction.__init__r-)r!r"r#r$rr%rrrrr.<s r.cs eZdZdZfddZZS)CustomArgumentGroupz Custom argument group that allows for the use of `-` or `_` in arguments passed and overrides the help for each when applicable. cst|}t|tjrt|d|d|d|d|d}n)t|tjr9t|d|d|d|d|d|d}n t|tjrFtdi|}t|}|S)Nrrr)r*r+r(r)vars isinstanceargparser.r&rr _add_action)r actionr rrrr3Ts$    zCustomArgumentGroup._add_action)r!r"r#r$r3r%rrrrr/Nsr/cs(eZdZdZfddZddZZS)CustomArgumentParserz Custom argument parser that allows for the use of `-` or `_` in arguments passed and overrides the help for each when applicable. cs<d|vr|ddkrt|d<nt|d<tj|i|dS)Nr4 store_true)r.rr add_argument)r r r rrrr7os  z!CustomArgumentParser.add_argumentcOs&t|g|Ri|}|j||Sr)r/_action_groupsr )r r r grouprrradd_argument_groupxs z'CustomArgumentParser.add_argument_group)r!r"r#r$r7r:r%rrrrr5is  r5) r2Actionrr&r._ArgumentGroupr/ArgumentParserr5rrrrs