adjust API for robustness of KL methods