Increase run logging (#470)

mystic-ai · Jul 18, 2024 · 70d04a3 · 70d04a3
1 parent 4a9e279
commit 70d04a3
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/pipeline/container/manager.py b/pipeline/container/manager.py
@@ -249,4 +249,5 @@ def run(
                 raise
             except Exception as exc:
                 raise RunnableError(exception=exc, traceback=traceback.format_exc())
+            logger.info("Run successful")
             return result
diff --git a/pipeline/container/routes/v4/runs.py b/pipeline/container/routes/v4/runs.py
@@ -49,7 +49,7 @@ async def run(
         response_queue: asyncio.Queue = asyncio.Queue()
         execution_queue.put_nowait((run_create, response_queue))
         run_output = await response_queue.get()
-
+        logger.info("Returning run result")
         response_schema, response.status_code = _generate_run_result(run_output)
         return response_schema
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pipeline-ai"
-version = "2.4.1"
+version = "2.4.2"
 description = "Pipelines for machine learning workloads."
 authors = [
   "Paul Hetherington <ph@mystic.ai>",