feat: minor fix to parser

Author: Wermutton

examples/prompt_selection.py

@@ -14,7 +14,7 @@
 prompt_selection = PromptSelection(
     model=model, num_responses=10, diversity_measure=diversity_measure, selection="min"
 )
-selected_prompt, selected_diversity, info = prompt_selection.generate(
+selected_prompt, selected_diversity = prompt_selection.generate(
         [
             "At the end, say 'the answer is [put the concatenated word here]'.\nQuestion: Take the last letter of each word in \"Tal Evan Lesley Sidney\" and concatenate them..",
             "At the end, say 'the answer is [put the concatenated word here]'.\nQuestion: Concatenate the last letter of each word in \"Tal Evan Lesley Sidney\".",

langdiversity/parser/answer_extractor.py

@@ -30,7 +30,7 @@ def extract_math_answer(response: str):
         try: responses.append(eval(r))
         except: pass
 
-    return set(responses)
+    return frozenset(responses)
 
 
 def extract_multi_choice_answer(response: str):

setup.py

@@ -8,7 +8,7 @@
 setup(
     name='langdiversity',
     packages=find_packages(exclude=['tests']),
-    version='0.0.2',
+    version='0.0.3',
     description='A tool to elevate your language models with insightful diversity metrics.',
     long_description=long_description,  
     long_description_content_type="text/markdown",