| import enum | |
| from typing import Sequence, Type, TypeVar | |
| T = TypeVar("T", bound=enum.Enum) | |
| class StrEnumMeta(enum.EnumMeta): | |
| auto = enum.auto | |
| def from_str(self: Type[T], member: str) -> T: # type: ignore[misc] | |
| try: | |
| return self[member] | |
| except KeyError: | |
| # TODO: use `add_suggestion` from torchvision.prototype.utils._internal to improve the error message as | |
| # soon as it is migrated. | |
| raise ValueError(f"Unknown value '{member}' for {self.__name__}.") from None | |
| class StrEnum(enum.Enum, metaclass=StrEnumMeta): | |
| pass | |
| def sequence_to_str(seq: Sequence, separate_last: str = "") -> str: | |
| if not seq: | |
| return "" | |
| if len(seq) == 1: | |
| return f"'{seq[0]}'" | |
| head = "'" + "', '".join([str(item) for item in seq[:-1]]) + "'" | |
| tail = f"{'' if separate_last and len(seq) == 2 else ','} {separate_last}'{seq[-1]}'" | |
| return head + tail | |