lml/ost#341: Add multiple cohort support.