Correcting test example

regression_evaluate.py

@@ -18,6 +18,7 @@ import datasets
 import re
 from statistics import median
 import numpy as np
+import ast
 
 # TODO: Add BibTeX citation
 _CITATION = """\
@@ -51,7 +52,7 @@ Examples:
     >>> metric = evaluate.load("rfr2003/regression_evaluate")
     >>> results = metric.compute(generations=['[150, 0]'], golds=[183, 177, 146, 85, 70, 78, 55, 17, 0, -1, -1])
     >>> print(results)
-    {'precision': 4.0, 'recall': 345.0, 'macro-mean': 174.5, 'median macro-mean': 174.5}
+    {'precision': 4.0, 'recall': 344.0, 'macro-mean': 174.0, 'median macro-mean': 174.0}
 """
 
 
@@ -104,8 +105,10 @@ class regression_evaluate(evaluate.Metric):
     
         for gen, gold in zip(generations, golds):
             f_gold = list(set([float(g) for g in gold]))
-     
-            f_ans = re.findall(r'\d+(?:\.\d+)?', gen)
+            try:
+                f_ans = ast.literal_eval(gen)
+            except:
+                f_ans = re.findall(r'\d+(?:\.\d+)?', gen)
                 
             f_ans = list(set([float(a) for a in f_ans])) #get rid of duples values
     

tests.py

@@ -2,6 +2,6 @@ test_cases = [
     {
         'generations': ['[150, 0]'],
         'golds': [183, 177, 146, 85, 70, 78, 55, 17, 0, -1, -1],
-        "result": {'precision': 4.0, 'recall': 345.0, 'macro-mean': 174.5, 'median macro-mean': 174.5}
+        "result": {'precision': 4.0, 'recall': 344.0, 'macro-mean': 174.0, 'median macro-mean': 174.0}
     }
 ]