mirror of https://github.com/JDAI-CV/fast-reid.git
14 lines
367 B
Python
14 lines
367 B
Python
import sys
|
|
|
|
class ExceptionHook:
|
|
instance = None
|
|
|
|
def __call__(self, *args, **kwargs):
|
|
if self.instance is None:
|
|
from IPython.core import ultratb
|
|
self.instance = ultratb.FormattedTB(mode='Plain',
|
|
color_scheme='Linux', call_pdb=1)
|
|
return self.instance(*args, **kwargs)
|
|
|
|
sys.excepthook = ExceptionHook()
|