That is a bit difficult now because I am using a function from `grid_search`. I have added a hack that will be there as long as the boolean indices in cv generators.