Displaying azimuthal information with a cyclic colormap

Hello,

I am working with 3D vector data, and I want to show some stream plots with the color corresponding to the azimuth angle. I can use the calculator and atan2 to calculate the azimuth [-pi, pi], and with cyclic colormap this looks alright in most areas. The issue is that the stream plot interpolates between -pi and pi, the value is ~0, meaning the color is incorrect.

Is there a way to alter the stream plot interpolation parameters? Or effectively include periodic boundary conditions on the azimuth?

Thanks