def reward_fn(samples):
    "Gives a negative count of rooms for each sample"
    return [-sample.count(":") for sample in samples]
Gives a negative count of rooms for each sample
promptist/trlx/examples/architext.py