From aba26633f792ed16920e311c5e625f5d2f3b9182 Mon Sep 17 00:00:00 2001
From: Nolan <nboukachab@teklia.com>
Date: Tue, 30 Aug 2022 13:36:31 +0000
Subject: [PATCH] Display updated model version
---
arkindex_worker/worker/training.py | 13 +++++++++++--
1 file changed, 11 insertions(+), 2 deletions(-)
diff --git a/arkindex_worker/worker/training.py b/arkindex_worker/worker/training.py
index 2501e0eb..e3beb8cc 100644
--- a/arkindex_worker/worker/training.py
+++ b/arkindex_worker/worker/training.py
@@ -145,6 +145,9 @@ class TrainingMixin(object):
"description": description,
},
)
+ logger.info(
+ f"Model version ({model_version_details['id']}) was created successfully"
+ )
except ErrorResponse as e:
if e.status_code >= 500:
logger.error(f"Failed to create model version: {e.content}")
@@ -158,6 +161,10 @@ class TrainingMixin(object):
logger.error(model_version_details[0])
return
+ logger.info(
+ f"Model version ({model_version_details['id']}) has the same hash, using this one instead of creating one"
+ )
+
return model_version_details
def upload_to_s3(self, archive_path: str, model_version_details: dict) -> None:
@@ -185,11 +192,12 @@ class TrainingMixin(object):
Update the specified model version to the state `Available` and use the given information"
"""
- logger.info("Updating the model version...")
+ model_version_id = model_version_details.get("id")
+ logger.info(f"Updating model version ({model_version_id})")
try:
self.request(
"UpdateModelVersion",
- id=model_version_details.get("id"),
+ id=model_version_id,
body={
"state": "available",
"description": model_version_details.get("description"),
@@ -197,5 +205,6 @@ class TrainingMixin(object):
"tag": model_version_details.get("tag"),
},
)
+ logger.info(f"Model version ({model_version_id}) was successfully updated")
except ErrorResponse as e:
logger.error(f"Failed to update model version: {e.content}")
--
GitLab