remove print in api

paddle_fl/core/scheduler/agent_master.py

@@ -104,10 +104,7 @@ class FLScheduler(object):
 
     def start_fl_training(self):
         # loop until training is done
-        loop = 0
         while True:
-            if loop <= 1:
-                print(loop)
             random.shuffle(self.fl_workers)
             worker_dict = {}
             for worker in self.fl_workers[:self.sample_worker_num]:

paddle_fl/core/trainer/diffiehellman/diffiehellman.py

@@ -43,8 +43,6 @@ try:
     rng = RAND_bytes
 except(AttributeError, ImportError):
     rng = os.urandom
-    #raise RNGError
-#python2
 
 class DiffieHellman:
     """

paddle_fl/core/trainer/fl_trainer.py

@@ -90,12 +90,12 @@ class FLTrainer(object):
         # TODO(guru4elephant): add connection with master
         if self.cur_step != 0:
             while not self.agent.finish_training():
-                print('wait others finish')
+                self._logger.debug("Wait others finish")
                 continue
         while not self.agent.can_join_training():
-            print("wait permit")
+            self._logger.debug("Wait permit")
             continue
-        print("ready to train")
+        self._logger.debug("Ready to train")
         return False
 
 
@@ -124,7 +124,6 @@ class FedAvgTrainer(FLTrainer):
         self.exe.run(self._recv_program)
         epoch = 0
         for i in range(num_epoch):
-                print(epoch)
                 for data in reader():
                     self.exe.run(self._main_program,
                            feed=feeder.feed(data),