Fix documentation in partition.py

There are probably other places, but at any rate we need this extra line for it to look correct.