Add __getstate__ and __setstate__ for IterableQueue (#74)

cgarciae · Jan 5, 2022 · 00b0e22 · 00b0e22
1 parent 5231806
commit 00b0e22
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pypeln/process/queue.py b/pypeln/process/queue.py
@@ -112,6 +112,13 @@ def get_pipeline_exception(self, exception: BaseException) -> PipelineException:
 
         return PipelineException(exception_type, trace)
 
+    def __getstate__(self):
+        return super().__getstate__() + (self.namespace, self.exception_queue)
+
+    def __setstate__(self, state):
+        super().__setstate__(state[:-2])
+        self.namespace, self.exception_queue = state[-2:]
+
 
 class OutputQueues(tp.List[IterableQueue[T]], tp.Generic[T]):
     def put(self, x: T):