diff --git a/arkindex_worker/worker/training.py b/arkindex_worker/worker/training.py index 00dbd09e37dacd45ad127323918d8decc19b9d4a..6750e1f00a18da600720fa6877ef5d70c594d154 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}")