Simplify the API, remove ensure_partition so get_partition will create by default as well