Skip to content

Commit

Permalink
small cleaning
Browse files Browse the repository at this point in the history
  • Loading branch information
Andreas Hellander committed Dec 6, 2023
1 parent ecd5714 commit 104cc48
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 6 deletions.
4 changes: 2 additions & 2 deletions fedn/fedn/common/storage/models/memorymodelstorage.py
Original file line number Diff line number Diff line change
Expand Up @@ -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})
4 changes: 2 additions & 2 deletions fedn/fedn/common/storage/models/modelstorage.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Expand Down
4 changes: 2 additions & 2 deletions fedn/fedn/common/storage/models/tempmodelstorage.py
Original file line number Diff line number Diff line change
Expand Up @@ -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})

Expand Down

0 comments on commit 104cc48

Please sign in to comment.