⚡ perf: improve user experience with clearer log messages

zrr-lab · Oct 19, 2024 · 0c20b5d · 0c20b5d
1 parent c6a4aa7
commit 0c20b5d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/nanoflow/task.py b/nanoflow/task.py
@@ -60,7 +60,7 @@ async def wrapper_fn():
                 logger.error(f"Failed to execute task: {e}")
                 if retry_interval:
                     retry = retry_interval.pop(0)
-                    logger.info(f"Retry task after {retry} seconds")
+                    logger.info(f"Retry task `{self.name}` after {retry} seconds")
                     await asyncio.sleep(retry)
                     return await wrapper_fn()
                 raise e

diff --git a/nanoflow/utils.py b/nanoflow/utils.py
@@ -76,7 +76,7 @@ def inner_fn() -> None:
                 process = subprocess.Popen(command, shell=True, env=environ)
             returncode = process.wait()
             if returncode != 0:
-                raise TaskProcessError(f"Task failed with return code {returncode}")
+                raise TaskProcessError(f"Task `{name}` failed with return code {returncode}")
 
         return inner_fn