Fix/modelrepository get model repo (#195)

* fix: change get_model_repo func to normal func instead of class method

* build: update version to 2.1.5

* fix: remove log

h1st/model/model.py

@@ -47,6 +47,7 @@ def __init__(self):
         self.stats = {}
         self.metrics = {}
         self.base_model = None
+        self.model_repo = ModelRepository()
 
     def persist(self, version=None) -> str:
         """
@@ -58,15 +59,15 @@ def persist(self, version=None) -> str:
         :param version: model version, leave blank for autogeneration
         :returns: model version
         """
-        repo = ModelRepository.get_model_repo(self)
+        repo = self.model_repo.get_model_repo(self)
         return repo.persist(model=self, version=version)
 
     def load(self, version: str = None) -> Any:
         """
         Load parameters from the specified `version` from the ModelRepository.
         Leave version blank to load latest version.
         """
-        repo = ModelRepository.get_model_repo(self)
+        repo = self.model_repo.get_model_repo(self)
         repo.load(model=self, version=version)
 
         return self

h1st/model/repository/model_repository.py

@@ -437,7 +437,6 @@ def persist(self, model, version=None):
             logger.exception(f'Error persisting model {model} version {version}: {e}')
         finally:
             dir_util.remove_tree(tmpdir)
-            logger.info(f'Removed temp dir {tmpdir}')
 
         return version
 
@@ -526,15 +525,14 @@ def _get_key(self, model, version):
 
         return key
 
-    @classmethod
-    def get_model_repo(cls, ref=None):
+    def get_model_repo(self, ref=None):
         """
         Retrieve the default model repository for the project
 
         :param ref: target model
         :returns: Model repository instance
         """
-        if not hasattr(cls, "MODEL_REPO"):  # global ModelRepository.MODEL_REPO
+        if not hasattr(self, "MODEL_REPO"):  # ModelRepository.MODEL_REPO
             repo_path = None
             if ref is not None:
                 # root module
@@ -568,9 +566,9 @@ def get_model_repo(cls, ref=None):
             if not repo_path:
                 raise RuntimeError("Please set MODEL_REPO_PATH in config.py")
 
-            setattr(cls, "MODEL_REPO", ModelRepository(storage=repo_path))
+            self.MODEL_REPO = ModelRepository(storage=repo_path)
 
-        return getattr(cls, "MODEL_REPO")
+        return self.MODEL_REPO
 
 
 def _tar_create(target, source):

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "h1st"
-version = "2.1.4"
+version = "2.1.5"
 description = "Human-First AI (H1st)"
 authors = ["Aitomatic, Inc. <[email protected]>"]
 license = "Apache-2.0"