EDFA has a gain spectrum, and the gain for each wavelength needs to be calculated. But if you want to skip all the calculations and just input the gain, you can use “EDFA black box” component in the same library. You can also use “Optical Amplifier” component, by which you can define parameters such as gain, noise figure,… without having to design amplifier’s other parameters so that it gives the aforementioned parameters.