Skip to content

Commit 51efc22

Browse files
authored
Updated Report class in examples to adapt the change in the simple.Report class. (#152)
* Fixed attribute error in fedsarah_client.py * Update report attribute in axiothea_client.py
1 parent d48ed41 commit 51efc22

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

examples/axiothea/axiothea_client.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -29,4 +29,4 @@ async def train(self):
2929
report, weights = await super().train()
3030

3131
return Report(report.num_samples, report.accuracy,
32-
report.training_time, report.data_loading_time), weights
32+
report.training_time, report.update_response), weights

examples/fedsarah/fedsarah_client.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ async def train(self):
4848
fn = f"new_client_control_variates_{self.client_id}.pth"
4949
os.remove(fn)
5050
return Report(report.num_samples, report.accuracy,
51-
report.training_time, report.data_loading_time,
51+
report.training_time, report.update_response,
5252
2), [weights, self.client_control_variates]
5353

5454
def load_payload(self, server_payload):

0 commit comments

Comments
 (0)