diff --git a/image_prediction/formatter/formatters/enum.py b/image_prediction/formatter/formatters/enum.py index a3c6cc6..b679279 100644 --- a/image_prediction/formatter/formatters/enum.py +++ b/image_prediction/formatter/formatters/enum.py @@ -1,9 +1,6 @@ from enum import Enum from image_prediction.formatter.formatters.key_formatter import KeyFormatter -from image_prediction.utils import get_logger - -logger = get_logger() class EnumFormatter(KeyFormatter): @@ -11,5 +8,4 @@ class EnumFormatter(KeyFormatter): return key.value if isinstance(key, Enum) else key def transform(self, obj): - logger.debug("EnumFormatter.transform") raise NotImplementedError diff --git a/src/serve.py b/src/serve.py index feff704..b749a85 100644 --- a/src/serve.py +++ b/src/serve.py @@ -27,12 +27,6 @@ def main(): if __name__ == "__main__": logging.basicConfig(level=CONFIG.service.logging_level) - # logging.getLogger("flask").setLevel(logging.ERROR) - # logging.getLogger("urllib3").setLevel(logging.ERROR) - # logging.getLogger("werkzeug").setLevel(logging.ERROR) - # logging.getLogger("waitress").setLevel(logging.ERROR) - # logging.getLogger("PIL").setLevel(logging.ERROR) - # logging.getLogger("h5py").setLevel(logging.ERROR) show_banner() diff --git a/test/integration_tests/actual_server_test.py b/test/integration_tests/actual_server_test.py index 4255893..31e8431 100644 --- a/test/integration_tests/actual_server_test.py +++ b/test/integration_tests/actual_server_test.py @@ -3,7 +3,7 @@ from multiprocessing import Process import pytest import requests -from funcy import retry, compose +from funcy import retry from waitress import serve from image_prediction.flask import make_prediction_server @@ -38,7 +38,6 @@ def server_type(request): @pytest.fixture def server(server_type): - if server_type == "dummy": return make_prediction_server(lambda x: int(x.decode()) // 2)