o ]i@s>dZddlZddlZddlmZGdddejZddZdS)z4Custom parser to override tuples in the config dict.N)flagsc@s eZdZdZddZddZdS) TupleParseraJParser for tuple arguments. Custom flag parser for Tuple objects that is based on the existing parsers in `absl.flags`. This parser can be used to read in and override tuple arguments. It outputs a `tuple` object from an existing `tuple` or `str`. The ony requirement is that the overriding parameter should be a `tuple` as well. The overriding parameter can have a different number of elements of different type than the original. For a detailed list of what `str` arguments are supported for overriding, look at `ast.literal_eval` from the Python Standard Library. cCs2t|tr|St|trt|S|durdS|fS)aReturns a `tuple` representing the input `argument`. Args: argument: The argument to be parsed. Valid types are `tuple` and `str` or a single object. `str` arguments are parsed and converted to a tuple, a single object is converted to a tuple of length 1, and an empty `tuple` is returned for arguments `NoneType`. Returns: A `TupleType` representing the input argument as a `tuple`. N) isinstancetuplestr_convert_str_to_tuple)selfargumentrr\/home/ubuntu/.local/lib/python3.10/site-packages/ml_collections/config_flags/tuple_parser.pyparse#s  zTupleParser.parsecCsdS)Nrr)r rrr flag_type8szTupleParser.flag_typeN)__name__ __module__ __qualname____doc__r r rrrr rs rc Cszt|}WnAty|}Yn7tyH}z+d|vr$d|vr$|}nd|dtt|d}d|vr9|d7}t||WYd}~nd}~wwt|trP|St|tjj r`t|t s`t|S|fS)zFunction to convert a Python `str` object to a `tuple`. Args: string: The `str` to be converted. Returns: A `tuple` version of the string. Raises: ValueError: If the string is not a well formed `tuple`. , zError while parsing string: z` as tuple. If you intended to pass the argument as a single element, use quotes such as `--flag z., otherwise insert quotes around each element.zD Use commas instead of whitespace as the separator between elements.N) ast literal_eval ValueError SyntaxErrorreprrr collectionsabcIterabler)stringvalueexcmsgrrr r<s4     r) rrcollections.abcrabslrArgumentParserrrrrrr s   &