Update partitioning.py documentation.