Lomb Scargle Algorithm for mutiple frequencies

I am working on implementing the Astropy Lomb Scargle Periodogram to estimate the coefficients on a spectrally-dependent modulation function, which is the sum of a sin and cos wave. For now I am working with simulated data. The modulation function is unevenly sampled in wavenumber, which is why the LSP was selected for data reduction. The frequency of the modulation function is also dependent on wavenumber. I have a working example of the LSP for a constant frequency, but I need the algorithm to work for an array of frequencies, since the frequency changes with wavenumber. Please see my example on my lab’s github:

If anyone can help me with this, I would really appreciate it. Thank you!