Digitos que suman 17 MPI

from mpi4py import MPI
comm = MPI.COMM_WORLD
rank = comm.Get_rank()
size = comm.Get_size()

def suman17(n):
    digitos = list(str(n))
    suma = 0
    for d in digitos:
        suma += int(d)
    return suma == 17

def primeroMayor(n):
    digitos = list(str(n))
    return int(digitos[0]) > int(digitos[-1])

def digitos2y6(n):
    digitos = list(str(n))
    return "2" in digitos and "6" in digitos

numeros = []
for n in range(0, 100_000, size):
    if suman17(n) and primeroMayor(n) and digitos2y6(n):
        numeros.append(n)

suma_parciales = sum(numeros)
sumas_parciales = comm.gather(suma_parciales)
if rank == 0:
        print(sum(sumas_parciales))