fix bug in serving_api

PaddlePaddle · Sep 9, 2020 · fd96a1e · fd96a1e
1 parent c6416f8
commit fd96a1e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/paddle_fl/paddle_fl/core/trainer/fl_trainer.py b/python/paddle_fl/paddle_fl/core/trainer/fl_trainer.py
@@ -19,6 +19,7 @@
 from paddle.fluid.io import is_belong_to_optimizer
 from paddle_fl.paddle_fl.core.scheduler.agent_master import FLWorkerAgent
 import numpy
+import paddle_serving_client.io as serving_io
 import hmac
 import hashlib
 from .diffiehellman.diffiehellman import DiffieHellman