Fix inverted probability logic in RandomExpand.__call__

[pantment]

This PR fixes a bug in RandomExpand.call where the probability check was inverted. According to the docstring, prob is the probability to expand, but the previous implementation expanded with probability 1 - prob. The condition has been corrected so that expansion now happens with the intended probability.

[CLAassistant]

All committers have signed the CLA.