o 7t¾iã@sHddlmZmZdedeefdd„Zdedeefdd„Zdd „Zd S) é)ÚDictÚListÚdocÚreturncCs t|dƒS)z.Return all of the accepted answers as choices.Úanswers)Ú_remove_prefixes)r©rúM/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/webqs/utils.pyÚ doc_to_choices r cCst|dƒ}ttt|ƒƒƒS)z9Return list of indices of accepted answers (all of them).r)rÚlistÚrangeÚlen)rÚ remainingrrr Ú doc_to_target s rcCs@| ¡|dg}|dd…D]}| |d¡s| |¡q|S)zÁ Remove any alias that has a strict prefix elsewhere in the list. This is an optimization. We can do this because if the prefix is acceptable by isgreedy, we can stop looking. réNéÿÿÿÿ)ÚsortÚ startswithÚappend)ÚaliasesÚretÚaliasrrr rs  €rN)ÚtypingrrÚstrr Úintrrrrrr Ús