Fix not all exceptions being able to be pickled (#543)

src/distilabel/__init__.py

@@ -13,9 +13,7 @@
 # limitations under the License.
 
 from rich import traceback as rich_traceback
-from tblib import pickling_support
 
 __version__ = "1.0.0"
 
 rich_traceback.install(show_locals=True)
-pickling_support.install()

src/distilabel/pipeline/local.py

@@ -20,6 +20,8 @@
 import traceback
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union, cast
 
+import tblib
+
 from distilabel.distiset import create_distiset
 from distilabel.llms.mixins import CudaDevicePlacementMixin
 from distilabel.pipeline.base import BasePipeline, _Batch, _BatchManager, _WriteBuffer
@@ -488,6 +490,9 @@ def _error_callback(self, e: BaseException) -> None:
             with self.shared_info[_STEPS_LOADED_LOCK_KEY]:
                 self.shared_info[_STEPS_LOADED_KEY] = [_STEPS_LOADED_ERROR_CODE]
             _SUBPROCESS_EXCEPTION = e.subprocess_exception
+            _SUBPROCESS_EXCEPTION.__traceback__ = tblib.Traceback.from_string(
+                e.formatted_traceback
+            ).as_traceback()
             return
 
         # If the step is global, is not in the last trophic level and has no successors,
@@ -598,7 +603,9 @@ def __init__(
         self.step = step
         self.code = code
         self.subprocess_exception = subprocess_exception
-        self.formatted_traceback = traceback.format_exc()
+        self.formatted_traceback = "".join(
+            traceback.format_exception(subprocess_exception)
+        )
 
     @classmethod
     def create_load_error(