Apply colored tracebacks on a wider scale

GreyHy · Jan 20, 2023 · dcda989 · dcda989
1 parent c37c860
commit dcda989
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/seleniumbase/__init__.py b/seleniumbase/__init__.py
@@ -9,6 +9,7 @@
 from seleniumbase.__version__ import __version__
 from seleniumbase.common import decorators  # noqa
 from seleniumbase.common import encryption  # noqa
+from seleniumbase.core import colored_traceback
 from seleniumbase.core.browser_launcher import get_driver  # noqa
 from seleniumbase.fixtures import js_utils  # noqa
 from seleniumbase.fixtures import page_actions  # noqa
@@ -34,6 +35,7 @@
         pdb.DefaultConfig.enable_hidden_frames = False
         pdb.DefaultConfig.truncate_long_lines = True
         pdb.DefaultConfig.sticky_by_default = True
+colored_traceback.add_hook()
 if sys.version_info >= (3, 7):
     webdriver.TouchActions = None  # Lifeline for past selenium-wire versions
 if sys.version_info >= (3, 10):

diff --git a/seleniumbase/plugins/sb_manager.py b/seleniumbase/plugins/sb_manager.py
@@ -111,7 +111,6 @@ def SB(
     from seleniumbase import BaseCase
     from seleniumbase import config as sb_config
     from seleniumbase.config import settings
-    from seleniumbase.core import colored_traceback
     from seleniumbase.fixtures import constants
     from seleniumbase.fixtures import shared_utils
 
@@ -770,7 +769,6 @@ def SB(
         sb.headless_active = False
     test_name = None
     terminal_width = shared_utils.get_terminal_width()
-    colored_traceback.add_hook()
     if test:
         import colorama
         import os