I don't understand the if and the comment. If you want not random valid and test set, should you force the rng to be None for valid and test. For train, if it is None, create it? It seam it miss a not in the if.

I don't know MultiplexingMatrixIterator. How to make it return not randomized order? rng=None?