Update sklearn_utils.py