17 lines
485 B
Python
17 lines
485 B
Python
from funcy import rcompose
|
|
|
|
from image_prediction.transformer.transformer import Transformer
|
|
from image_prediction.utils import get_logger
|
|
|
|
logger = get_logger()
|
|
|
|
|
|
class TransformerCompositor(Transformer):
|
|
def __init__(self, formatter: Transformer, *formatters: Transformer):
|
|
formatters = (formatter, *formatters)
|
|
self.pipe = rcompose(*formatters)
|
|
|
|
def transform(self, obj):
|
|
logger.debug("TransformerCompositor.transform")
|
|
return self.pipe(obj)
|