Fixed CriticalChildException in TaskManager.py (#737)

openwpm · Aug 25, 2020 · 5d06a6b · 5d06a6b
1 parent 5b2d271
commit 5d06a6b
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/automation/TaskManager.py b/automation/TaskManager.py
@@ -346,14 +346,8 @@ def _check_failure_status(self) -> None:
                 "failure limit.", self.failure_status['CommandSequence']
             )
         if self.failure_status['ErrorType'] == 'CriticalChildException':
-            exc = pickle.loads(self.failure_status['Exception'])
-            assert type(exc) == BaseException, (
-                'Unexpected object passed in place of exception while handling'
-                ' a critical exception in a child process. Please report this '
-                'error to https://github.com/mozilla/OpenWPM/issues/547. '
-                f'Object was of type {type(exc)} and looked like {exc!r}.'
-            )
-            raise exc
+            exc_type, exc, tb = pickle.loads(self.failure_status['Exception'])
+            raise exc.with_traceback(tb)
 
     # CRAWLER COMMAND CODE