Skip to content

Commit e216139

Browse files
authored
update data context (#86)
* Update data_context.py
1 parent 0e76301 commit e216139

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

python/wedpr_ml_toolkit/wedpr_ml_toolkit/context/data_context.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,9 @@ def to_psi_format(self, merge_filed, result_receiver_id_list):
2828
dataset_psi = []
2929
for dataset in self.datasets:
3030
if dataset.agency in result_receiver_id_list:
31-
result_receiver = "true"
31+
result_receiver = True
3232
else:
33-
result_receiver = "false"
33+
result_receiver = False
3434
dataset_psi_info = {"idFields": [merge_filed],
3535
"dataset": {"owner": dataset.dataset_owner,
3636
"ownerAgency": dataset.agency,
@@ -45,13 +45,13 @@ def to_model_formort(self, merge_filed, result_receiver_id_list):
4545
dataset_model = []
4646
for dataset in self.datasets:
4747
if dataset.agency in result_receiver_id_list:
48-
result_receiver = "true"
48+
result_receiver = True
4949
else:
50-
result_receiver = "false"
50+
result_receiver = False
5151
if dataset.is_label_holder:
52-
label_provider = "true"
52+
label_provider = True
5353
else:
54-
label_provider = "false"
54+
label_provider = False
5555
dataset_psi_info = {"idFields": [merge_filed],
5656
"dataset": {"owner": dataset.dataset_owner,
5757
"ownerAgency": dataset.agency,

0 commit comments

Comments
 (0)