A trick to partition the ruptures by trt_id