o }oi*@s ddlmZGdddeZdS))Enumc@s*eZdZdZddZedefddZdS) PrettyStrEnuma Pretty enum to work with string values for config options with choices Provides en automatic error message with possible values, if the value is not in the enum Converting to string will show the actual string value, which makes serialization/deserialization straightforward Example: class ASRModelType(PrettyStrEnum): CTC = "ctc" RNNT = "rnnt" ... model_type = ModelType(model_type_string) # automatically validated if model_type == ModelType.CTC: # more error-prone (to typos) compared to pure string literals ... # do something specific to CTC model cCs|jS)N)value)selfrC/home/ubuntu/.local/lib/python3.10/site-packages/nemo/utils/enum.py__str__"szPrettyStrEnum.__str__rcCs*dtt|}t|d|jd|)Nz, z is not a valid z. Possible choices: )joinmapstr ValueError__name__)clsrchoicesrrr _missing_%szPrettyStrEnum._missing_N)r __module__ __qualname____doc__r classmethodobjectrrrrrrs rN)enumrrrrrrs