o .wih@sddlmZddlmZGdddeZGdddeZGdddeZGd d d eZGd d d eZGd ddeZ GdddeZ dS))StrEnum)Literalc sReZdZdZedefddZed deddede d ddffd d Z Z S) EnumStrz Base Enum.returncCdS)NTaskrrrY/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/torchmetrics/utilities/enums.py_namez EnumStr._namekeyclsvaluesource)r ranyc stztj|dd|d}W||Sty9}zdd|jD}td|d||d|d |d }~ww) 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. -_)rcSsg|]}|qSr)lower).0mrrr 0sz$EnumStr.from_str..zInvalid z: expected one of z , but got .N)superfrom_strreplace ValueError_member_names_r _allowed_matches)r rrmeerr _allowed_im __class__rr rs zEnumStr.from_str)r ) __name__ __module__ __qualname____doc__ staticmethodstrr classmethodtyperr __classcell__rrr!r rs 2rc@s2eZdZdZedefddZdZdZdZ dZ d S) DataTypezOEnum to represent data type. >>> "Binary" in list(DataType) True rcCr)Nz Data typerrrrr r ?r zDataType._namebinaryz multi-labelz multi-classzmulti-dim multi-classN) r#r$r%r&r'r(r BINARY MULTILABEL MULTICLASSMULTIDIM_MULTICLASSrrrr r,7sr,c@s6eZdZdZedefddZdZdZdZ dZ d Z dS) AverageMethodzEnum to represent average method. >>> None in list(AverageMethod) True >>> AverageMethod.NONE == None True >>> AverageMethod.NONE == 'none' True rcCr)NzAverage methodrrrrr r Ur zAverageMethod._namemicromacroweightedNsamples) r#r$r%r&r'r(r MICROMACROWEIGHTEDNONESAMPLESrrrr r2Is r2c@*eZdZdZedefddZdZdZdS)MDMCAverageMethodz7Enum to represent multi-dim multi-class average method.rcCr)NzMDMC Average methodrrrrr r cr zMDMCAverageMethod._nameglobal samplewiseN) r#r$r%r&r'r(r GLOBAL SAMPLEWISErrrr r=`s r=c@s.eZdZdZedefddZdZdZdZ dS) ClassificationTaskz}Enum to represent the different tasks in classification metrics. >>> "binary" in list(ClassificationTask) True rcCrNClassificationrrrrr r sr zClassificationTask._namer- multiclass multilabelN) r#r$r%r&r'r(r r.r0r/rrrr rBksrBc@r<)ClassificationTaskNoBinaryzEnum to represent the different tasks in classification metrics. >>> "binary" in list(ClassificationTaskNoBinary) False rcCrrCrrrrr r r z ClassificationTaskNoBinary._namerFrEN) r#r$r%r&r'r(r r/r0rrrr rG| rGc@r<)ClassificationTaskNoMultilabelzEnum to represent the different tasks in classification metrics. >>> "multilabel" in list(ClassificationTaskNoMultilabel) False rcCrrCrrrrr r r z$ClassificationTaskNoMultilabel._namer-rEN) r#r$r%r&r'r(r r.r0rrrr rIrHrIN) lightning_utilities.core.enumsrtyping_extensionsrrr,r2r=rBrGrIrrrr s  $