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 index 3f8bffb75..a97c4534c 100755 --- 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 index b4a2427d4..4254cfb26 100755 --- 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 index e5bd820e4..9be136aa8 100644 --- 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