ihddlmZddlmZGddeZGddeZGddeZGd d eZGd d eZGd deZ GddeZ y))StrEnum)Literalc beZdZdZedefdZed deddede dddffd Z xZ S) EnumStrz Base Enum.returncy)NTaskr q/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/torchmetrics/utilities/enums.py_namez EnumStr._namesr clsvaluesource)keyranyc : t||jdd|}||S#t$re}|jDcgc]}|j ncc}w}}td|j d|j|d|d|d}~wwxYw) 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. -_)rzInvalid z: expected one of z , but got .N)superfrom_strreplace ValueError_member_names_lowerr _allowed_matches)rrrmeerrm _allowed_im __class__s r rzEnumStr.from_strs& !%--S"9&!IB 2w  .1.@.@A.@1779.@AKA399;-'9#:N:Nv:V9WWabgahhij  s !, BBA;BB)r) __name__ __module__ __qualname____doc__ staticmethodstrr classmethodtyperr __classcell__)r"s@r rrsZ3d9oc7CX;Yfor rc6eZdZdZedefdZdZdZdZ dZ y) DataTypezOEnum to represent data type. >>> "Binary" in list(DataType) True rcy)Nz Data typer r r r r zDataType._name?sr binaryz multi-labelz multi-classzmulti-dim multi-classN) r#r$r%r&r'r(r BINARY MULTILABEL MULTICLASSMULTIDIM_MULTICLASSr r r r-r-7s93FJJ1r r-c:eZdZdZedefdZdZdZdZ dZ dZ y) AverageMethodzEnum to represent average method. >>> None in list(AverageMethod) True >>> AverageMethod.NONE == None True >>> AverageMethod.NONE == 'none' True rcy)NzAverage methodr r r r r zAverageMethod._nameUr micromacroweightedNsamples) r#r$r%r&r'r(r MICROMACROWEIGHTEDNONESAMPLESr r r r5r5Is=  3   E EH DGr r5c.eZdZdZedefdZdZdZy)MDMCAverageMethodz7Enum to represent multi-dim multi-class average method.rcy)NzMDMC Average methodr r r r r zMDMCAverageMethod._namecs$r global samplewiseN) r#r$r%r&r'r(r GLOBAL SAMPLEWISEr r r rBrB`s*A%3%%FJr rBc2eZdZdZedefdZdZdZdZ y)ClassificationTaskz}Enum to represent the different tasks in classification metrics. >>> "binary" in list(ClassificationTask) True rcyNClassificationr r r r r zClassificationTask._namesr7r r/ multiclass multilabelN) r#r$r%r&r'r(r r0r2r1r r r rIrIks3 3  FJJr rIc.eZdZdZedefdZdZdZy)ClassificationTaskNoBinaryzEnum to represent the different tasks in classification metrics. >>> "binary" in list(ClassificationTaskNoBinary) False rcyrKr r r r r z ClassificationTaskNoBinary._namer7r rNrMN) r#r$r%r&r'r(r r1r2r r r rPrP|s. 3  JJr rPc.eZdZdZedefdZdZdZy)ClassificationTaskNoMultilabelzEnum to represent the different tasks in classification metrics. >>> "multilabel" in list(ClassificationTaskNoMultilabel) False rcyrKr r r r r z$ClassificationTaskNoMultilabel._namer7r r/rMN) r#r$r%r&r'r(r r0r2r r r rSrSs. 3  FJr rSN) lightning_utilities.core.enumsrtyping_extensionsrrr-r5rBrIrPrSr r r rWsb3%!g!H2w2$G."   W r