From 104cc48377ce0b437289ea9d58175c261ffa650b Mon Sep 17 00:00:00 2001 From: Andreas Hellander Date: Wed, 6 Dec 2023 14:59:42 +0100 Subject: [PATCH] small cleaning --- fedn/fedn/common/storage/models/memorymodelstorage.py | 4 ++-- fedn/fedn/common/storage/models/modelstorage.py | 4 ++-- fedn/fedn/common/storage/models/tempmodelstorage.py | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/fedn/fedn/common/storage/models/memorymodelstorage.py b/fedn/fedn/common/storage/models/memorymodelstorage.py index 4f9bdfd83..70a01f37b 100644 --- a/fedn/fedn/common/storage/models/memorymodelstorage.py +++ b/fedn/fedn/common/storage/models/memorymodelstorage.py @@ -41,10 +41,10 @@ def get_ptr(self, model_id): """ return self.models[model_id] - def get_meta(self, model_id): + def get_model_metadata(self, model_id): return self.models_metadata[model_id] - def set_meta(self, model_id, model_metadata): + def set_model_metadata(self, model_id, model_metadata): self.models_metadata.update({model_id: model_metadata}) diff --git a/fedn/fedn/common/storage/models/modelstorage.py b/fedn/fedn/common/storage/models/modelstorage.py index c15f47c87..4945ab37b 100644 --- a/fedn/fedn/common/storage/models/modelstorage.py +++ b/fedn/fedn/common/storage/models/modelstorage.py @@ -26,7 +26,7 @@ def get(self, model_id): pass @abstractmethod - def get_meta(self, model_id): + def get_model_metadata(self, model_id): """ Get model metadata from storage :param model_id: The model id @@ -37,7 +37,7 @@ def get_meta(self, model_id): pass @abstractmethod - def set_meta(self, model_id, model_metadata): + def set_model_metadata(self, model_id, model_metadata): """ Set model metadata in storage :param model_id: The model id diff --git a/fedn/fedn/common/storage/models/tempmodelstorage.py b/fedn/fedn/common/storage/models/tempmodelstorage.py index 295f23278..93d8c458c 100644 --- a/fedn/fedn/common/storage/models/tempmodelstorage.py +++ b/fedn/fedn/common/storage/models/tempmodelstorage.py @@ -62,11 +62,11 @@ def get_ptr(self, model_id): self.models[model_id] = {'file': f} return self.models[model_id]['file'] - def get_meta(self, model_id): + def get_model_metadata(self, model_id): return self.models_metadata[model_id] - def set_meta(self, model_id, model_metadata): + def set_model_metadata(self, model_id, model_metadata): self.models_metadata.update({model_id: model_metadata})