diff --git a/flytekit/__init__.py b/flytekit/__init__.py index 047bbf2674..d8def79a42 100644 --- 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)