Skip to content

Commit

Permalink
it runs !
Browse files Browse the repository at this point in the history
  • Loading branch information
jeandut committed Jan 15, 2024
1 parent 7e63934 commit 196c5c6
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion fedeca/strategies/bootstraper.py
Original file line number Diff line number Diff line change
Expand Up @@ -283,6 +283,7 @@ def local_computation(self, datasamples, shared_state=None) -> list:
res = getattr(self, name_decorated_function)(
datasamples=bootstrapped_data, shared_state=shared_state[idx], _skip=True
)

self.checkpoints_list[idx] = copy.deepcopy(self._get_state_to_save())

# We restore the algo to its old state
Expand Down Expand Up @@ -346,7 +347,7 @@ def aggregation(self, shared_states=None) -> list:
name_decorated_function = aggregation_function.__name__ + "_original"
if shared_states is not None:
# loop over the aggregation steps provided using _skip=True
for shared_state in shared_states:
for shared_state in zip(*shared_states):
res = getattr(self, name_decorated_function)(shared_states=shared_state, _skip=True)
results.append(res)
else:
Expand Down

0 comments on commit 196c5c6

Please sign in to comment.