Skip to content

Commit b3208aa

Browse files
authored
Fix NVFLARE demo (dmlc#8340)
1 parent 748d516 commit b3208aa

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

demo/nvflare/custom/trainer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@ def _do_training(self, fl_ctx: FLContext):
5353
client_name = fl_ctx.get_prop(FLContextKey.CLIENT_NAME)
5454
rank = int(client_name.split('-')[1]) - 1
5555
communicator_env = {
56+
'xgboost_communicator': 'federated',
5657
'federated_server_address': self._server_address,
5758
'federated_world_size': self._world_size,
5859
'federated_rank': rank,

0 commit comments

Comments
 (0)