diff --git a/fleet_rec/core/trainers/cluster_trainer.py b/fleet_rec/core/trainers/cluster_trainer.py
index 06f3a603db5b59db5cbb84a2f83aa2feca475ca2..4d47a4e051f44cd4a21168752cbfdd18cd119a65 100644
--- a/fleet_rec/core/trainers/cluster_trainer.py
+++ b/fleet_rec/core/trainers/cluster_trainer.py
@@ -103,7 +103,7 @@ class ClusterTrainer(TranspileTrainer):
         self._exe.run(fleet.startup_program)
         context['status'] = 'train_pass'
 
-   def dataloader_train(self, context):
+    def dataloader_train(self, context):
         fleet.init_worker()
 
         reader = self._get_dataloader()
diff --git a/fleet_rec/core/trainers/tdm_cluster_trainer.py b/fleet_rec/core/trainers/tdm_cluster_trainer.py
index 2b87b3c84cf068784a2f10d864007de6782cf6da..40a46d92fd7aae583c313be9e27865d700b6ca73 100644
--- a/fleet_rec/core/trainers/tdm_cluster_trainer.py
+++ b/fleet_rec/core/trainers/tdm_cluster_trainer.py
@@ -19,6 +19,7 @@ Training use fluid with one node only.
 
 from __future__ import print_function
 import logging
+import numpy as np
 import paddle.fluid as fluid
 from paddle.fluid.incubate.fleet.parameter_server.distribute_transpiler import fleet
 from paddle.fluid.incubate.fleet.parameter_server.distribute_transpiler.distributed_strategy import StrategyFactory