Integration with sklearn hyperparameter optimisation