How to use map with methods

map_method.py

# The following is Steven Bethard's functions to pickle methods - required to
# use multiprocessing.Pool with model.run

import copy_reg
import types

def _pickle_method(method):
    func_name = method.im_func.__name__
    obj = method.im_self
    cls = method.im_class
    return _unpickle_method, (func_name, obj, cls)

def _unpickle_method(func_name, obj, cls):
    for cls in cls.mro():
        try:
            func = cls.__dict__[func_name]
        except KeyError:
            pass
        else:
            break
    return func.__get__(obj, cls)

copy_reg.pickle(types.MethodType, _pickle_method, _unpickle_method)

# Now define class and run Pool.map() on MyClass.do()

class MyClass(object):

    def __init__(self):
        pass

    def do(self, a):
        return a * a

from multiprocessing import Pool
p = Pool()
m = MyClass()
results = p.map(m.do, range(10))

print results