16 lines
409 B
Python
16 lines
409 B
Python
from enum import Enum
|
|
|
|
from image_prediction.formatter.formatters.key_formatter import KeyFormatter
|
|
from image_prediction.utils import get_logger
|
|
|
|
logger = get_logger()
|
|
|
|
|
|
class EnumFormatter(KeyFormatter):
|
|
def format_key(self, key):
|
|
return key.value if isinstance(key, Enum) else key
|
|
|
|
def transform(self, obj):
|
|
logger.debug("EnumFormatter.transform")
|
|
raise NotImplementedError
|