def METHOD_NAME(rng_type):
    # Helper to avoid consuming rngs
    if rng_type == "none":
        return None
    elif rng_type == "int":
        return 42
    else:
        return np.random.RandomState(0)