All parameters in aggregation

[felix-marx]

Hi everyone,
I was running the examples and implementing a custom aggregation.
However, I only receive the parameters for each layer (also separated into weights and bias) individually and not all the parameters from every collaborator during the aggregation.
Instead, the aggregation function is called multiple times to aggregate all parameters.
This makes it impossible to implement aggregations like FLTrust as they require all parameters.
Is there a setting I need to change in one of the YAML files or is this not possible with this framework?

I was following the example given in Federated_Pytorch_MNIST_custom_aggregation_Tutorial and implemented the custom aggregation using the AggregationFunction interface.