a  dV@sddlmZddlmZGdddeZGdddeZGdddeZGd d d eZGd d d eZGd ddeZ GdddeZ dS))StrEnum)LiteralcsFeZdZdZeedddZed eedddfdd Z Z S) EnumStrz Base Enum.returncCsdS)NZTaskrrre/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torchmetrics/utilities/enums.py_namesz EnumStr._namekey)r valueany)r sourcerc sztj|dd|d}Wn\tyx}zDdd|jD}td|d||d|d |WYd }~n d }~00||S) aLoad from string. Raises: ValueError: If required value is not among the supported options. >>> class MyEnum(EnumStr): ... a = "aaa" ... b = "bbb" >>> MyEnum.from_str("a") >>> MyEnum.from_str("c") Traceback (most recent call last): ... ValueError: Invalid Task: expected one of ['a', 'b'], but got c. -_)r cSsg|] }|qSr)lower).0mrrr /z$EnumStr.from_str..zInvalid z: expected one of z , but got .N)superfrom_strreplace ValueError_member_names_r Z_allowed_matches)clsr r meerrZ _allowed_im __class__rrrs zEnumStr.from_str)r ) __name__ __module__ __qualname____doc__ staticmethodstrr classmethodrr __classcell__rrrrrs rc@s2eZdZdZeedddZdZdZdZ dZ d S) DataTypezOEnum to represent data type. >>> "Binary" in list(DataType) True rcCsdS)Nz Data typerrrrrr >szDataType._namebinaryz multi-labelz multi-classzmulti-dim multi-classN) r r!r"r#r$r%r BINARY MULTILABEL MULTICLASSZMULTIDIM_MULTICLASSrrrrr(6sr(c@s6eZdZdZeedddZdZdZdZ dZ d Z dS) AverageMethodzEnum to represent average method. >>> None in list(AverageMethod) True >>> AverageMethod.NONE == None True >>> AverageMethod.NONE == 'none' True rcCsdS)NzAverage methodrrrrrr TszAverageMethod._namemicromacroZweightedNZsamples) r r!r"r#r$r%r ZMICROZMACROZWEIGHTEDNONEZSAMPLESrrrrr-Hs r-c@s*eZdZdZeedddZdZdZdS)MDMCAverageMethodz7Enum to represent multi-dim multi-class average method.rcCsdS)NzMDMC Average methodrrrrrr bszMDMCAverageMethod._nameglobalZ samplewiseN) r r!r"r#r$r%r GLOBALZ SAMPLEWISErrrrr1_s r1c@s.eZdZdZeedddZdZdZdZ dS) ClassificationTaskz}Enum to represent the different tasks in classification metrics. >>> "binary" in list(ClassificationTask) True rcCsdSNZClassificationrrrrrr rszClassificationTask._namer) multiclass multilabelN) r r!r"r#r$r%r r*r,r+rrrrr4js r4c@s*eZdZdZeedddZdZdZdS)ClassificationTaskNoBinaryzEnum to represent the different tasks in classification metrics. >>> "binary" in list(ClassificationTaskNoBinary) False rcCsdSr5rrrrrr sz ClassificationTaskNoBinary._namer7r6N) r r!r"r#r$r%r r+r,rrrrr8{s r8c@s*eZdZdZeedddZdZdZdS)ClassificationTaskNoMultilabelzEnum to represent the different tasks in classification metrics. >>> "multilabel" in list(ClassificationTaskNoMultilabel) False rcCsdSr5rrrrrr sz$ClassificationTaskNoMultilabel._namer)r6N) r r!r"r#r$r%r r*r,rrrrr9s r9N) Zlightning_utilities.core.enumsrZtyping_extensionsrrr(r-r1r4r8r9rrrrs  $