nit

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Oct 24, 2024 · 386b3e5 · 386b3e5
1 parent 8cc9737
commit 386b3e5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flytekit/clis/sdk_in_container/utils.py b/flytekit/clis/sdk_in_container/utils.py
@@ -121,13 +121,13 @@ def pretty_print_traceback(e: Exception, verbosity: int = 1):
         new_tb = remove_unwanted_traceback_frames(tb, unwanted_module_names)
         console.print(Traceback.from_exception(type(e), e, new_tb))
     elif verbosity == 1:
-        new_tb = remove_unwanted_traceback_frames(e.__traceback__, unwanted_module_names)
-        console.print(Traceback.from_exception(type(e), e, new_tb))
         click.secho(
             f"Frames from the following modules were removed from the traceback: {unwanted_module_names}."
             f" For more verbose output, use the flags -vv or -vvv.",
             fg="yellow",
         )
+        new_tb = remove_unwanted_traceback_frames(e.__traceback__, unwanted_module_names)
+        console.print(Traceback.from_exception(type(e), e, new_tb))
     elif verbosity >= 2:
         console.print(Traceback.from_exception(type(e), e, e.__traceback__))
     else: