which function sends information to the aggregation function?

[dskhanirfan]

what is the name of the function that is run on the participating collaborators which sends information to the aggregation function? Can you outline the function calls in a sequential manner i-e one function call after the other that leads to final aggregation in an FL round?

[ayyaz-azeem]

which function dispatches parameters run on individual collaborators to the aggregator function? @tanertopal @argretzi @kshannon

[itrushkin]

Hi @dskhanirfan! Thank you for your interest in OpenFL!

We perform aggregation at the end of the round. After all collaborators reported their last task results to aggregator, we start aggregation.
Sequence:

1. Collaborator.do_task() - Local task execution.
2. Collaborator.send_task_results() - Send the results to collaborator. Probably this is the method you are looking for.
3. AggregatorGRPCClient.send_local_task_results() - Wrap the data into gRPC request.
4. AggregatorGRPCServer.SendLocalTaskResults() - Unwrap the request.
5. Aggregator.send_local_task_results() - Process task results.
6. Aggregator._end_of_task_check() - Check whether task is completed on all collaborators.
7. Aggregator._end_of_round_check() - Check whether the task was the last task in round.
8. Aggregator._compute_validation_related_task_metrics() - Preparing the data for aggregation.
9. TensorDB.get_aggregated_tensor() - The aggregation itself. On line 171 aggregation_function is called.

[dskhanirfan]

Thank you @itrushkin can you kindly mention the filenames which contain the functions?

[itrushkin]

You can click on each method in the list above to find its location in code. Locations:

Steps 1-2: openfl/component/collaborator/collaborator.py
3: openfl/transport/grpc/aggregator_client.py
4: openfl/transport/grpc/aggregator_server.py
5-8: openfl/component/aggregator/aggregator.py
9: openfl/databases/tensor_db.py