adjust the API of KL methods for model selection

make the API work for `GradientModelSelection`
@karlnapf 
please merge this.