diff --git a/image_prediction/default_objects.py b/image_prediction/default_objects.py index 2bb8cb6..16ad466 100644 --- a/image_prediction/default_objects.py +++ b/image_prediction/default_objects.py @@ -7,7 +7,7 @@ from image_prediction.estimator.adapter.adapter import EstimatorAdapter from image_prediction.extractor_classifier.extractor_classifier import ExtractorClassifier from image_prediction.formatter.formatters.camel_case import Snake2CamelCaseKeyFormatter from image_prediction.formatter.formatters.enum import EnumFormatter -from image_prediction.formatter.formatters.response import ResponseTransformer +from image_prediction.transformer.transformers.response import ResponseTransformer from image_prediction.image_extractor.extractors.parsable import ParsablePDFImageExtractor from image_prediction.label_mapper.mappers.probability import ProbabilityMapper from image_prediction.model_loader.loader import ModelLoader diff --git a/image_prediction/formatter/formatters/response.py b/image_prediction/transformer/transformers/response.py similarity index 98% rename from image_prediction/formatter/formatters/response.py rename to image_prediction/transformer/transformers/response.py index 122dea8..06eee4e 100644 --- a/image_prediction/formatter/formatters/response.py +++ b/image_prediction/transformer/transformers/response.py @@ -8,6 +8,7 @@ from image_prediction.utils import get_logger logger = get_logger() +# TODO: move to transformers class ResponseTransformer(Transformer): def transform(self, data): logger.debug("ResponseTransformer.transform")