import style

intel-analytics · Aug 24, 2022 · b3fb235 · b3fb235
1 parent be89076
commit b3fb235
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/ppml/src/bigdl/ppml/fl/nn/fl_server.py b/python/ppml/src/bigdl/ppml/fl/nn/fl_server.py
@@ -17,13 +17,13 @@
 from concurrent import futures
 import grpc
 from bigdl.ppml.fl import *
-from bigdl.ppml.fl.nn.generated.nn_service_pb2_grpc import *
+
 from bigdl.ppml.fl.nn.nn_service import NNServiceImpl
 import yaml
 
 from ..psi.psi_service import PSIServiceImpl
-
-from .generated.psi_service_pb2_grpc import add_PSIServiceServicer_to_server
+from .generated.nn_service_pb2_grpc import *
+from .generated.psi_service_pb2_grpc import *