Modified docstring of `partition_all` to @mrocklin's suggestion in #65