From 312a6ebab3bdc9827fe697a9e06b4ea15b13e71e Mon Sep 17 00:00:00 2001 From: Tuan Pham Date: Thu, 29 Jun 2023 12:20:01 +0700 Subject: [PATCH] Fix/add persist log (#192) * feat: add log when persist model * build: update version to 2.1.2 --- h1st/model/repository/model_repository.py | 5 +++++ pyproject.toml | 2 +- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/h1st/model/repository/model_repository.py b/h1st/model/repository/model_repository.py index 56c79db5..3be906a0 100644 --- a/h1st/model/repository/model_repository.py +++ b/h1st/model/repository/model_repository.py @@ -410,6 +410,7 @@ def persist(self, model, version=None): # assert isinstance(model, Model) # TODO: use version format: v_20200714-1203 version = version or str(ulid.new()) + logger.info(f"Persisting model {model} version {version}") try: # serialize a model to a temporary folder and then clean up later @@ -420,21 +421,25 @@ def persist(self, model, version=None): self._serder.serialize(model, serialized_dir) _tar_create(tar_file, serialized_dir) + logger.info(f'Created tar file {tar_file} at {serialized_dir} for model {model} version {version}') with open(tar_file, mode="rb") as f: self._storage.set_bytes( self._get_key(model, version), f.read(), ) + logger.info(f'Set bytes {self._get_key(model, version)} to storage from {tar_file} - {serialized_dir}') self._storage.set_obj( self._get_key(model, "latest"), version, ) + logger.info(f'Set obj {self._get_key(model, "latest")} to storage from {tar_file} - {serialized_dir}') model.version = version finally: dir_util.remove_tree(tmpdir) + logger.info(f'Removed temp dir {tmpdir}') return version diff --git a/pyproject.toml b/pyproject.toml index 49c423c5..3448057b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "h1st" -version = "2.1.1" +version = "2.1.2" description = "Human-First AI (H1st)" authors = ["Aitomatic, Inc. "] license = "Apache-2.0"