o W+ iq@s`ddlmZmZmZddlmZddlmZGdddeZGdddeZ Gdd d e Z d S) )ActionArgumentDefaultsHelpFormatterArgumentParser)fields)ListcsXeZdZdZdfdd ZddZ  ddeffdd Zdfd d Zd d Z Z S)CliArgumentParserz Argument Parser to define and parse command-line args for training. Args: training_args: dict or list of dict which defines different parameters for training. Nc s4d|vrt|d<tjdi|||_|dS)Nformatter_class)rsuper__init__ training_args define_args)selfr kwargs __class__r s/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/modelscope/trainers/cli_argument_parser.pyr s  zCliArgumentParser.__init__cCsdd|DS)NcSs"g|] }|dr|ddqS)--N) startswith).0argr r r s"z5CliArgumentParser.get_manual_args..r )rargsr r rget_manual_argssz!CliArgumentParser.get_manual_argsrcsX|dur|jnd|_d|vr||dd|_|||_tj||g|Ri|S)Nz--model)modelZmodel_idindexrZ manual_argsr _parse_known_args)rr namespaceZ args_extrarrr rrs z#CliArgumentParser._parse_known_argscs t|SN)r print_help)rfilerr rr!$s zCliArgumentParser.print_helpcCs|jdurCt|jD]:}|j}t|j|j}d|}t|j|jdd}||d<d|jvr5|jd|d<t|d<|j|fi|q dSdS)Nrhelp)typer#defaultchoicesaction) r rnamegetattrdictr$metadata SingleAction add_argument)rfZarg_nameZarg_attrr(rr r rr 's   zCliArgumentParser.define_argsr )NN) __name__ __module__ __qualname____doc__r rrrr!r __classcell__r r rrrs rc@s0eZdZdZeddZeddZddZdS) DictActionaR argparse action to split an argument into KEY=VALUE form on the first = and append to a dictionary. List options can be passed as comma separated values, i.e 'KEY=V1,V2,V3', or with explicit brackets, i.e. 'KEY=[V1,V2,V3]'. It also support nested brackets to build list/tuple values. e.g. 'KEY=[(V1,V2),(V3,V4)]' cCsdzt|WStyYnwzt|WStyYnw|dvr*|dkS|dkr0dS|S)N)truefalser5None)int ValueErrorfloatlower)valr r rparse_int_float_bool_str@s      z#DictAction.parse_int_float_bool_strcCsdd}|ddd}d}|dr"|dr"d }|d d }n|d r3|d r3|d d }n d|vr>> DictAction._parse_iterable('1,2,3') [1, 2, 3] >>> DictAction._parse_iterable('[a, b, c]') ['a', 'b', 'c'] >>> DictAction._parse_iterable('[(1, 2, 3), [a, b], c]') [(1, 2, 3), ['a', 'b'], 'c'] cSs|d|dkr|d|dksJd|t|}t|D]'\}}|d|}|dkrJ|d|dkrJ|d|dkrJ|}|Sq#|S)aFind the position of next comma in the string. If no ',' is found in the string, return the string length. All chars inside '()' and '[]' are treated as one element and thus ',' inside these brackets are ignored. ()[]zImbalanced brackets exist in N,)countlen enumerate)stringendidxcharprer r rfind_next_commaas  z2DictAction.parse_iterable..find_next_commaz'" Fr>r?Trr@rArBrN) stripreplacerendswithr4r=rDparse_iterableappendtuple)r<rKis_tuplevaluesZ comma_idxelementr r rrRPs(    zDictAction.parse_iterablec Cs@i}|D]}|jddd\}}||||<qt||j|dS)N=r)maxsplit)splitrRsetattrdest) rparserrrV option_stringoptionskvkeyr<r r r__call__s zDictAction.__call__N)r/r0r1r2 staticmethodr=rRrbr r r rr47s   9r4c@seZdZdZddZdS)r,z Argparse action to convert value to tuple or list or nested structure of list and tuple, i.e 'V1,V2,V3', or with explicit brackets, i.e. '[V1,V2,V3]'. It also support nested brackets to build list/tuple values. e.g. '[(V1,V2),(V3,V4)]' cCs4t|trt||j||dSt||j|dSr ) isinstancestrr[r\rR)rr]rvaluer^r r rrbs zSingleAction.__call__N)r/r0r1r2rbr r r rr,s r,N) argparserrr dataclassesrtypingrrr4r,r r r rs   1[