From 999ff63087f6ed6e831efd3f618a6412b641fe6f Mon Sep 17 00:00:00 2001 From: barrierye Date: Sun, 5 Jul 2020 02:41:01 +0800 Subject: [PATCH] fix check for response_op --- python/pipeline/pipeline_server.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/pipeline/pipeline_server.py b/python/pipeline/pipeline_server.py index c65e889c..b849a4e5 100644 --- a/python/pipeline/pipeline_server.py +++ b/python/pipeline/pipeline_server.py @@ -20,7 +20,7 @@ from contextlib import closing import yaml from .proto import pipeline_service_pb2_grpc -from .operator import Op +from .operator import ResponseOp from .profiler import TimeProfiler from .dag import DAGExecutor @@ -49,8 +49,8 @@ class PipelineServer(object): pass def set_response_op(self, response_op): - if not isinstance(response_op, Op): - raise Exception("response_op must be Op type.") + if not isinstance(response_op, ResponseOp): + raise Exception("response_op must be ResponseOp type.") if len(response_op.get_input_ops()) != 1: raise Exception("response_op can only have one previous op.") self._response_op = response_op -- GitLab