提交 68feafd7 编写于 作者: B barrierye

fix test_pipeline_server.py

上级 afb05e6b
...@@ -30,7 +30,7 @@ logging.basicConfig( ...@@ -30,7 +30,7 @@ logging.basicConfig(
class ImdbRequestOp(RequestOp): class ImdbRequestOp(RequestOp):
def load_user_resources(self): def init_op(self):
self.imdb_dataset = IMDBDataset() self.imdb_dataset = IMDBDataset()
self.imdb_dataset.load_resource('imdb.vocab') self.imdb_dataset.load_resource('imdb.vocab')
...@@ -46,7 +46,7 @@ class ImdbRequestOp(RequestOp): ...@@ -46,7 +46,7 @@ class ImdbRequestOp(RequestOp):
class CombineOp(Op): class CombineOp(Op):
def preprocess(self, input_data): def preprocess(self, input_data, private_obj):
combined_prediction = 0 combined_prediction = 0
for op_name, data in input_data.items(): for op_name, data in input_data.items():
_LOGGER.info("{}: {}".format(op_name, data["prediction"])) _LOGGER.info("{}: {}".format(op_name, data["prediction"]))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册