How to use mpi4py with msgpack serialization

mpi4py_msgpack_demo.py

#!/usr/bin/env python

"""
How to use mpi4py with msgpack serialization
"""

from mpi4py import MPI 
import msgpack

# mpi4py assumes that the serializer function takes more than one parameter:
MPI.pickle.dumps = lambda *x: msgpack.dumps(x[0])
MPI.pickle.loads = msgpack.loads

comm = MPI.COMM_WORLD
size = comm.Get_size()
rank = comm.Get_rank()

in_data = str(rank)
comm.send(in_data, ((rank+1) % size))
out_data = comm.recv(source=(rank-1) % size)

print 'node %s: received %s' % (rank, out_data)