Pyro.sample from RelaxedOneHotCategorical doesn't return sampled indexes

sampled_indexes = pyro.sample(f"{address}_{index}", pyro.distributions.RelaxedOneHotCategorical(1, char_dist), obs=observed[index]).squeeze(0)

I have this code where sampled_indexes becomes a tensor of probabilities over a categorical distribution of size 32 rather than just sampling an index, how do I get this to just sample an index instead?