Changed keyerror. Replaced index with end and entity with entity_group eea43d1 Shashank14 commited on Jul 28, 2022