Generalize the reading of probability metadata to work for both branches