Add logs

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