debug CI 3, re-rasie the exception when handle the data

dmlc · Jul 29, 2022 · c0b1a99 · c0b1a99
1 parent a3b1060
commit c0b1a99
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python-package/xgboost/core.py b/python-package/xgboost/core.py
@@ -450,6 +450,7 @@ def _handle_exception(self, fn: Callable, dft_ret: _T) -> _T:
             # On dask, the worker is restarted and somehow the information is
             # lost.
             self._exception = e.with_traceback(tb)
+            self.reraise()
         return dft_ret
 
     def reraise(self) -> None: