BAMA python mpi code