From 0f263e48140fa78843446be7b6e28a6036618ad7 Mon Sep 17 00:00:00 2001
From: NolanB <nboukachab@teklia.com>
Date: Tue, 30 Aug 2022 15:03:36 +0200
Subject: [PATCH] Add logs
---
arkindex_worker/worker/training.py | 11 +++++++----
1 file changed, 7 insertions(+), 4 deletions(-)
diff --git a/arkindex_worker/worker/training.py b/arkindex_worker/worker/training.py
index 00dbd09e..6750e1f0 100644
--- a/arkindex_worker/worker/training.py
+++ b/arkindex_worker/worker/training.py
@@ -145,6 +145,8 @@ class TrainingMixin(object):
"description": description,
},
)
+ model_version_id = model_version_details.get("id")
+ logger.info(f"Model version ({model_version_id}) was created successfully")
except ErrorResponse as e:
if e.status_code >= 500:
logger.error(f"Failed to create model version: {e.content}")
@@ -158,8 +160,9 @@ class TrainingMixin(object):
logger.error(model_version_details[0])
return
- model_version_id = model_version_details.get("id")
- logger.info(f"Model version{model_version_id} was created successfully")
+ model_version_id = model_version_details["id"]
+ logger.info(f"Model version ({model_version_id}) already exist")
+
return model_version_details
def upload_to_s3(self, archive_path: str, model_version_details: dict) -> None:
@@ -188,7 +191,7 @@ class TrainingMixin(object):
"""
model_version_id = model_version_details.get("id")
- logger.info(f"Updating the model version {model_version_id}")
+ logger.info(f"Updating model version ({model_version_id})")
try:
self.request(
"UpdateModelVersion",
@@ -200,6 +203,6 @@ class TrainingMixin(object):
"tag": model_version_details.get("tag"),
},
)
- logger.info(f"Model version{model_version_id} successfully updated")
+ 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