Mixing data from multiple sources

HI!
I use data from multiple sources and I need to mix it with certain proportions. Each datasource is a directory with *.tfrecord files. I’ve thought about writing custom datasource or an external mixer class. What is an optimal way of doing this with ray.data?