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 515fd352483e1a3ced57014e11d4faff2a6faddb..92a15379c0b6ae1ad0cdc1401a01556e41c7eed7 100644 --- a/python/examples/pipeline/imdb_model_ensemble/test_pipeline_server.py +++ b/python/examples/pipeline/imdb_model_ensemble/test_pipeline_server.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. # pylint: disable=doc-string-missing -import paddle_serving_server.pipeline as pipeline from paddle_serving_server.pipeline import Op, RequestOp, ResponseOp from paddle_serving_server.pipeline import PipelineServer from paddle_serving_server.pipeline.proto import pipeline_service_pb2 @@ -22,12 +21,12 @@ from paddle_serving_app.reader import IMDBDataset import logging _LOGGER = logging.getLogger() -console_handler = pipeline.logger.StreamHandler() -console_handler.setLevel(logging.INFO) -console_handler.setFormatter( +user_handler = logging.StreamHandler() +user_handler.setLevel(logging.INFO) +user_handler.setFormatter( logging.Formatter( "%(levelname)s %(asctime)s [%(filename)s:%(lineno)d] %(message)s")) -_LOGGER.addHandler(console_handler) +_LOGGER.addHandler(user_handler) class ImdbRequestOp(RequestOp):