From 68feafd72bb6db1419ee5aa1c6d2f632e2e08221 Mon Sep 17 00:00:00 2001 From: barrierye Date: Sun, 5 Jul 2020 01:03:14 +0800 Subject: [PATCH] fix test_pipeline_server.py --- .../pipeline/imdb_model_ensemble/test_pipeline_server.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/examples/pipeline/imdb_model_ensemble/test_pipeline_server.py b/python/examples/pipeline/imdb_model_ensemble/test_pipeline_server.py index bbd37de5..08c6ccce 100644 --- a/python/examples/pipeline/imdb_model_ensemble/test_pipeline_server.py +++ b/python/examples/pipeline/imdb_model_ensemble/test_pipeline_server.py @@ -30,7 +30,7 @@ logging.basicConfig( class ImdbRequestOp(RequestOp): - def load_user_resources(self): + def init_op(self): self.imdb_dataset = IMDBDataset() self.imdb_dataset.load_resource('imdb.vocab') @@ -46,7 +46,7 @@ class ImdbRequestOp(RequestOp): class CombineOp(Op): - def preprocess(self, input_data): + def preprocess(self, input_data, private_obj): combined_prediction = 0 for op_name, data in input_data.items(): _LOGGER.info("{}: {}".format(op_name, data["prediction"])) -- GitLab