pyre.dice.sample_categorial works on dicts and lists