Add compatibility with scikit-learn > 1.5 #major

Add compatibility with scikit-learn > 1.5 #major #902