diff --git a/fedn/network/combiner/aggregators/fedopt.py b/fedn/network/combiner/aggregators/fedopt.py index 61a5858d..5ce19384 100644 --- a/fedn/network/combiner/aggregators/fedopt.py +++ b/fedn/network/combiner/aggregators/fedopt.py @@ -133,8 +133,7 @@ def combine_models(self, helper=None, delete_models=True, parameters=None): nr_aggregated_models += 1 # Delete model from storage if delete_models: - self.update_handler.delete_model( - model_update.model_update_id) + self.update_handler.delete_model(model_update) logger.info("AGGREGATOR({}): Deleted model update {} from storage.".format( self.name, model_update.model_update_id)) except Exception as e: