[enh] Better traceback when in debug mode

Kastakin · Aug 3, 2023 · b980e12 · b980e12
1 parent 66ec54a
commit b980e12
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/main/python/pyes/workers.py b/src/main/python/pyes/workers.py
@@ -1,6 +1,7 @@
 import logging
 import os
 import time
+import traceback
 from datetime import datetime
 from pathlib import Path
 
@@ -55,7 +56,12 @@ def run(self):
         try:
             optimizer.fit(self.data)
         except Exception as e:
-            self.signals.aborted.emit(str(e))
+            if self.debug:
+                self.signals.aborted.emit(
+                    "".join(traceback.TracebackException.from_exception(e).format())
+                )
+            else:
+                self.signals.aborted.emit(str(e))
             return None
 
         self.signals.log.emit(r"DATA LOADED!")