Implemented the allow restart if complete flag for a partitioned step