From 045b1a88bc9c5b2f3640818ae4ee2a6dbdaffe5d Mon Sep 17 00:00:00 2001
From: "zuchen.wang" <zuchen.wang@vipshop.com>
Date: Mon, 11 Oct 2021 11:31:30 +0800
Subject: [PATCH] fix import error in fastclas add bughook.py

---
 .gitignore                             |  2 ++
 projects/FastClas/fastclas/__init__.py |  2 +-
 tools/bughook.py                       | 13 +++++++++++++
 tools/train_net.py                     |  2 ++
 4 files changed, 18 insertions(+), 1 deletion(-)
 create mode 100644 tools/bughook.py

diff --git a/.gitignore b/.gitignore
index 8be82c3..cc5a199 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,9 +1,11 @@
 
 logs
+datasets
 
 # compilation and distribution
 __pycache__
 _ext
+*.c
 *.pyc
 *.pyd
 *.so
diff --git a/projects/FastClas/fastclas/__init__.py b/projects/FastClas/fastclas/__init__.py
index cfabb8c..38f40ad 100644
--- a/projects/FastClas/fastclas/__init__.py
+++ b/projects/FastClas/fastclas/__init__.py
@@ -5,6 +5,6 @@
 """
 
 from .bee_ant import *
-from .distracted_driver import *
+# from .distracted_driver import *
 from .dataset import ClasDataset
 from .trainer import ClasTrainer
diff --git a/tools/bughook.py b/tools/bughook.py
new file mode 100644
index 0000000..765543d
--- /dev/null
+++ b/tools/bughook.py
@@ -0,0 +1,13 @@
+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()
diff --git a/tools/train_net.py b/tools/train_net.py
index 1a0a3c7..caf4ca5 100644
--- a/tools/train_net.py
+++ b/tools/train_net.py
@@ -9,6 +9,8 @@ import sys
 
 sys.path.append('.')
 
+import bughook
+
 from fastreid.config import get_cfg
 from fastreid.engine import DefaultTrainer, default_argument_parser, default_setup, launch
 from fastreid.utils.checkpoint import Checkpointer