From 5d188d118df43575d2f7dc76f5257a89d1622d08 Mon Sep 17 00:00:00 2001
From: Andreas Hellander <andreas.hellander@gmail.com>
Date: Mon, 16 Dec 2024 13:05:42 +0100
Subject: [PATCH] Fix bug in fedopt that causes model delete to fail in fedopt

---
 fedn/network/combiner/aggregators/fedopt.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

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: