Enrich your train fold with a custom sampler inside an imblearn pipeline