address comments

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Aug 12, 2024 · 89e61cc · 89e61cc
1 parent 33b53dd
commit 89e61cc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/flytekit/bin/entrypoint.py b/flytekit/bin/entrypoint.py
@@ -184,9 +184,8 @@ def get_traceback_str(e: Exception) -> str:
     tb = e.__cause__.__traceback__ if e.__cause__ else e.__traceback__
     lines = traceback.format_tb(tb)
     lines = [line.rstrip() for line in lines]
-    lines = "\n".join(lines).split("\n")
-    tb_str = "\n    ".join([""] + lines)
-    format_str = "Traceback (most recent call last):\n" "{traceback}\n" "\n" "Message:\n" "\n" "    {message}"
+    tb_str = "\n    ".join(lines)
+    format_str = "Traceback (most recent call last):\n" "\n    {traceback}\n" "\n" "Message:\n" "\n" "    {message}"
 
     value = e.value if isinstance(e, FlyteUserRuntimeException) else e
     return format_str.format(traceback=tb_str, message=f"{type(value)}: {value}")

diff --git a/flytekit/exceptions/user.py b/flytekit/exceptions/user.py
@@ -17,7 +17,7 @@ def __init__(self, exc_value):
         @param exc_value: The exception that was raised from user code.
         """
         self._exc_value = exc_value
-        super(FlyteUserRuntimeException, self).__init__(str(exc_value))
+        super().__init__(str(exc_value))
 
     @property
     def value(self):