Display updated model version

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}")