Improve partition_consumer docs formatting

Untested, still trying to figure out how to preview those changes in the resulting docs