joblib.Memory support in knn divs