In https://github.com/atong01/conditional-flow-matching/blob/main/examples/images/conditional_mnist.ipynb
The following line reorders the samples in the batch for both x0 and x1 followed by sampling trajectories
t, xt, ut = FM.sample_location_and_conditional_flow(x0, x1)
However the label y is still ordered according to x1 before reordering.

As a consequence the labels y are mismatched with xt, they do not follow the same order.