diff --git a/fedn/fedn/network/storage/statestore/mongostatestore.py b/fedn/fedn/network/storage/statestore/mongostatestore.py index 5e229707e..aa7671802 100644 --- a/fedn/fedn/network/storage/statestore/mongostatestore.py +++ b/fedn/fedn/network/storage/statestore/mongostatestore.py @@ -184,9 +184,7 @@ def set_latest_model(self, model_id, session_id=None): """ committed_at = datetime.now() - current_model = self.model.find_one({"key": "current_model"}) - parent_model = None # if session_id is set the it means the model is generated from a session @@ -555,7 +553,6 @@ def get_model_ancestors(self, model_id: str, limit: int): :return: List of model ancestors. :rtype: list """ - model = self.model.find_one({"key": "models", "model": model_id}) current_model_id = model["parent_model"] if model is not None else None result = []