From 613bba8cfcf06176e90b618422a962c83a405df9 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Sat, 2 Apr 2022 02:45:21 +0200 Subject: [PATCH] ... --- image_prediction/classifier/classifier.py | 2 +- image_prediction/formatter/formatters/camel_case.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/image_prediction/classifier/classifier.py b/image_prediction/classifier/classifier.py index 9b8bcee..b537aa1 100644 --- 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) diff --git a/image_prediction/formatter/formatters/camel_case.py b/image_prediction/formatter/formatters/camel_case.py index 01d011f..caa240a 100644 --- 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("_")