From d04e4235089a5dee2553af43d9548ff8af500263 Mon Sep 17 00:00:00 2001 From: Luke Barr Date: Mon, 13 May 2019 04:41:56 -0500 Subject: [PATCH] Don't pass multiprocess object on pickle --- utils/iter_utils.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/utils/iter_utils.py b/utils/iter_utils.py index 75fb21f..eb00c55 100644 --- a/utils/iter_utils.py +++ b/utils/iter_utils.py @@ -48,6 +48,11 @@ class SubprocessGenerator(object): def __iter__(self): return self + def __getstate__(self): + self_dict = self.__dict__.copy() + del self_dict['p'] + return self_dict + def __next__(self): if self.p == None: user_param = self.user_param