Merge pull request #435 from RelevanceAI/development

v0.31.2

ai_transform/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.31.1"
+__version__ = "0.31.2"
 
 from ai_transform.timer import Timer
 

ai_transform/workflow/context_manager.py

@@ -1,6 +1,4 @@
 import os
-import logging
-import requests
 
 from inspect import Traceback
 from typing import Dict, Any, List
@@ -10,14 +8,12 @@
 from ai_transform.dataset import dataset
 from ai_transform.operator import abstract_operator
 from ai_transform.engine import abstract_engine
-from ai_transform.logger import format_logging_info
-from ai_transform.api.wrappers import request_wrapper
-from ai_transform.errors import UserFacingError
 from ai_transform.logger import ic
 
-WORKFLOW_FAIL_MESSAGE = (
-    "Workflow processed {:.2f}%" + " of documents. This is less than the success threshold of {:.2f}%"
-)
+
+WORKFLOW_PROCESSED_MESSAGE = "Workflow processed {:.2f}%" + " of documents. "
+
+WORKFLOW_FAIL_MESSAGE = WORKFLOW_PROCESSED_MESSAGE + "This is less than the success threshold of {:.2f}%"
 
 
 class WorkflowContextManager:
@@ -194,6 +190,8 @@ def __exit__(self, exc_type: type, exc_value: BaseException, traceback: Tracebac
         user_errors = None
 
         if self.engine is not None:
+            self.addtional_message = WORKFLOW_PROCESSED_MESSAGE.format(100 * self.engine.success_ratio)
+
             regular_workflow_failed = self.engine.success_ratio < self.success_threshold
 
             if regular_workflow_failed: