From 0861e225426df7750ea6dfd6b3d523963193df82 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Sat, 2 Apr 2022 03:38:44 +0200 Subject: [PATCH] fixed pipeline not working with flask... model was loaded in external process, probably; known issue --- image_prediction/redai_adapter/model.py | 5 +++++ test/integration_tests/actual_server_test.py | 5 +++-- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/image_prediction/redai_adapter/model.py b/image_prediction/redai_adapter/model.py index 60edcc2..c5450f5 100644 --- a/image_prediction/redai_adapter/model.py +++ b/image_prediction/redai_adapter/model.py @@ -1,5 +1,9 @@ from funcy import rcompose +from image_prediction.utils import get_logger + +logger = get_logger() + class PredictionModelHandle: """Simplifies usage of ModelHandle instances for prediction purposes.""" @@ -11,4 +15,5 @@ class PredictionModelHandle: return self.__predict(*args, **kwargs) def __call__(self, *args, **kwargs): + logger.debug("PredictionModelHandle.predict") return self.predict(*args, **kwargs) diff --git a/test/integration_tests/actual_server_test.py b/test/integration_tests/actual_server_test.py index 0e7992e..4255893 100644 --- a/test/integration_tests/actual_server_test.py +++ b/test/integration_tests/actual_server_test.py @@ -7,6 +7,7 @@ from funcy import retry, compose from waitress import serve from image_prediction.flask import make_prediction_server +from image_prediction.pipeline import load_pipeline @pytest.fixture @@ -36,13 +37,13 @@ def server_type(request): @pytest.fixture -def server(server_type, pipeline): +def server(server_type): if server_type == "dummy": return make_prediction_server(lambda x: int(x.decode()) // 2) elif server_type == "actual": - return make_prediction_server(compose(list, pipeline)) + return make_prediction_server(lambda x: list(load_pipeline(verbose=True)(x))) else: raise ValueError(f"Unknown server type {server_type}.")