xla

37b85187 - quick refactor on _get_group_assignment (#5318)