o o’×i9 ã@sbddlmZddlmZmZGdd„deeƒZGdd„deƒZGdd„deƒZGd d „d eƒZ d S) é)ÚEnum)ÚOptionalÚUnionc@sTeZdZdZedededfdd„ƒZdeeddfde fdd „Z de fd d „Z dS) ÚEnumStraType of any enumerator with allowed comparison to string invariant to cases. Example: >>> class MyEnum(EnumStr): ... ABC = 'abc' >>> MyEnum.from_str('Abc') >>> {MyEnum.ABC: 123} {: 123} ÚvalueÚreturncCs>dd„t|ƒDƒ}|D]}| ¡| ¡krt||ƒSq dS)NcSsg|] }| d¡s|‘qS)Ú_)Ú startswith)Ú.0Ústatus©r úZ/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/torchmetrics/utilities/enums.pyÚ sz$EnumStr.from_str..)ÚdirÚlowerÚgetattr)ÚclsrÚstatusesÚstr r r Úfrom_strs ÿzEnumStr.from_strÚotherNcCs*t|tƒr|jnt|ƒ}|j ¡| ¡kS©N)Ú isinstancerrÚstrr)Úselfrr r r Ú__eq__&szEnumStr.__eq__cCs t|jƒSr)ÚhashÚname)rr r r Ú__hash__*s zEnumStr.__hash__) Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú classmethodrrrrÚboolrÚintrr r r r rs  rc@s eZdZdZdZdZdZdZdS)ÚDataTypezNEnum to represent data type. >>> "Binary" in list(DataType) True Úbinaryz multi-labelz multi-classzmulti-dim multi-classN)rr r!r"ÚBINARYÚ MULTILABELÚ MULTICLASSÚMULTIDIM_MULTICLASSr r r r r&0s r&c@s$eZdZdZdZdZdZdZdZdS)Ú AverageMethodz®Enum to represent average method. >>> None in list(AverageMethod) True >>> AverageMethod.NONE == None True >>> AverageMethod.NONE == 'none' True ÚmicroÚmacroÚweightedNÚsamples) rr r!r"ÚMICROÚMACROÚWEIGHTEDÚNONEÚSAMPLESr r r r r,=s r,c@seZdZdZdZdZdS)ÚMDMCAverageMethodz7Enum to represent multi-dim multi-class average method.ÚglobalÚ samplewiseN)rr r!r"ÚGLOBALÚ SAMPLEWISEr r r r r6Osr6N) ÚenumrÚtypingrrrrr&r,r6r r r r Ús