fix import error in fastclas

add bughook.py

.gitignore

@@ -1,9 +1,11 @@
 
 logs
+datasets
 
 # compilation and distribution
 __pycache__
 _ext
+*.c
 *.pyc
 *.pyd
 *.so

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

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()

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