pass in partition step parameter