(OpenMP) push color iteration back to python