...

2022-04-02 02:45:21 +02:00 · 2022-04-02 02:45:21 +02:00 · 613bba8cfc
commit 613bba8cfc
parent 5c23898280
2 changed files with 2 additions and 3 deletions
--- a/image_prediction/classifier/classifier.py
+++ b/image_prediction/classifier/classifier.py
@ -30,5 +30,5 @@ class Classifier:
        return list(self.__pipe(batch))

    def __call__(self, batch: np.array) -> List[str]:
-        logger.debug("Classifier.predicting")
+        logger.debug("Classifier.predict")
        return self.predict(batch)
--- a/image_prediction/formatter/formatters/camel_case.py
+++ b/image_prediction/formatter/formatters/camel_case.py
@ -2,8 +2,7 @@ from image_prediction.formatter.formatters.key_formatter import KeyFormatter


 class Snake2CamelCaseKeyFormatter(KeyFormatter):
-    @staticmethod
-    def format_key(key):
+    def format_key(self, key):

        if isinstance(key, str):
            head, *tail = key.split("_")