xla

4fac7ce5 - quick refactor on _get_group_assignment