diff --git a/arkindex_worker/worker/training.py b/arkindex_worker/worker/training.py index 2501e0eb3eb9cac7681e6ce1ede839830d9e4126..e3beb8cc9e8929a9f23856a3f645e5bfb8729907 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}")