It seems to be a common problem: stratify_by is there but partition_by is not, meaning that the two sets should be non-overlapping on the value of a specific variable, such as video_id or patient_id.