I’m not sure if that would be easily achievable without writing an own derived class.
You could split your Dataset as the first step and pass the Subset to your main Dataset with all transformations.
Would that work for you?