psi fix bug (bytedance#121)

zhangqixun · Jul 13, 2020 · 7fa02c0 · 7fa02c0
1 parent 10a325b
commit 7fa02c0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/deploy/scripts/data_join/run_psi_data_join_follower_worker.sh b/deploy/scripts/data_join/run_psi_data_join_follower_worker.sh
@@ -22,7 +22,7 @@ if [ "$((WORKER_NUM % 2))" -ne "0" ]; then
   echo "Error: WORKER_NUM should be the multiplies of 2."
   exit 1
 fi
-if [ $INDEX -le $((WORKER_NUM / 2)) ]; then
+if [ $INDEX -lt $((WORKER_NUM / 2)) ]; then
   psi_preprocessor_cmd=/app/deploy/scripts/rsa_psi/run_psi_preprocessor.sh &
   exec ${psi_preprocessor_cmd}
   echo "launched run psi preprocessor"

diff --git a/deploy/scripts/data_join/run_psi_data_join_leader_worker.sh b/deploy/scripts/data_join/run_psi_data_join_leader_worker.sh
@@ -22,7 +22,7 @@ if [ "$((WORKER_NUM % 2))" -ne "0" ]; then
   echo "Error: WORKER_NUM should be the multiplies of 2."
   exit 1
 fi
-if [ $INDEX -le $((WORKER_NUM / 2)) ]; then
+if [ $INDEX -lt $((WORKER_NUM / 2)) ]; then
   psi_signer_cmd="/app/deploy/scripts/rsa_psi/run_rsa_psi_signer.sh"
   exec ${psi_signer_cmd} &
   echo "launched psi signer"

diff --git a/fedlearner/data_join/cmd/rsa_psi_preprocessor_cli.py b/fedlearner/data_join/cmd/rsa_psi_preprocessor_cli.py
@@ -135,7 +135,7 @@
                 batch_size=args.process_batch_size,
                 max_flying_item=args.max_flying_item
             ),
-            raw_data_options=dj_pb.RawDataOptions(
+            input_raw_data=dj_pb.RawDataOptions(
                 raw_data_iter=args.raw_data_iter,
                 compressed_type=args.compressed_type,
                 read_ahead_size=args.read_ahead_size