我有两个文件:
x.py
class BF(object)
def __init__():
.
.
def add(self,z):
.
.
py
from y import BF
def FUNC((a,b,bf))
.
.
bf.add(x)
.
.
return bf
.
.
if __name__ == '__main__':
pool = multiprocessing.Pool(3)
for i in range(len(sl)):
bf_set.append(BF())
results = pool.map(FUNC,zip(sl, itertools.repeat(aa), bf_set))
我也试图在FUNC内定义BF,但是我得到了:
PicklingError: Can't pickle <type 'instancemethod'>: attribute lookup __builtin__.instancemethod failed
我已经阅读了一些有关相关问题的文章,但是它们在类中都有其pool.map(),因此解决方案无法应用于该问题(我想)。
任何想法?
最佳答案
我将基本上使用上面的内容,但是将其转换为工作代码。如果使用dill
,则序列化没有问题。我正在使用称为multiprocessing
的pathos.multiprocessing
叉,它使用dill
而不是pickle
。
>>> def FUNC((a,b,bf)):
... z = a+b
... bf.add(z)
... return bf
...
>>> class BF(object):
... def add(self, z):
... self.z += z
... def __init__(self):
... self.z = 0
...
>>> from pathos.multiprocessing import ProcessingPool as Pool
>>> pool = Pool()
>>>
>>> f = BF()
>>> f.add(1)
>>> f.z
1
>>>
>>> FUNC((0,1,f))
<__main__.BF object at 0x10d387f50>
>>>
>>> FUNC((0,1,f)).z
2
>>>
>>> sl = [BF() for i in range(10)]
>>> results = pool.map(FUNC, zip(range(len(sl)), range(len(sl)), sl))
>>> [bf.z for bf in results]
[0, 2, 4, 6, 8, 10, 12, 14, 16, 18]
这是可行的,因为
pathos
使用dill
,可以在python中序列化几乎所有内容。>>> import dill as pickle
>>> pickle.loads(pickle.dumps(bf.add))
<bound method BF.add of <__main__.BF object at 0x10d383950>>
>>> pickle.loads(pickle.dumps(BF.add))
<unbound method BF.add>
在以下位置获取
pathos
和dill
:https://github.com/uqfoundation