From 2d0545c928fd0e49f04cdd2fd9d7b6d8d01dc904 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Sun, 3 Apr 2022 04:31:50 +0200 Subject: [PATCH] refactoring --- image_prediction/default_objects.py | 2 +- .../formatters => transformer/transformers}/response.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) rename image_prediction/{formatter/formatters => transformer/transformers}/response.py (98%) 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")