Where is wrapped gaussian sampling done during noising?

[tsjain]

The only code where I see a loop over k is in ScoreMatchingNoisedAnglesDataset class.

I am not sure that I was able to find the use of the wrapped gaussian during generation of the noised samples. Can you point me to the relevant code where this is being done?