(OpenMP) push color iteration up to python