Don't pass multiprocess object on pickle (#261)

2019-05-14 02:30:23 -05:00
parent d6e8dde481
commit ba4e377d13
1 changed files with 5 additions and 0 deletions
--- 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