Allow disabling rich tracebacks via env var

Signed-off-by: Fabio Grätz <[email protected]>
flyteorg · Jun 15, 2023 · 37b9039 · 37b9039
1 parent 207d019
commit 37b9039
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/flytekit/__init__.py b/flytekit/__init__.py
@@ -192,6 +192,7 @@
    SourceCode
 
 """
+import os
 import sys
 from typing import Generator
 
@@ -303,4 +304,5 @@ def load_implicit_plugins():
 load_implicit_plugins()
 
 # Pretty-print exception messages
-traceback.install(width=None, extra_lines=0)
+if os.environ.get("FLYTE_SDK_RICH_TRACEBACKS") != "0":
+    traceback.install(width=None, extra_lines=0)