Update app/tapas.py

app/tapas.py

@@ -84,7 +84,7 @@ def show_answers(queries, answers, aggregation_predictions_string):
     Visualize the postprocessed answers.
   """
     agg = {"NONE": lambda x: x, "SUM" : lambda x: sum(x), "AVERAGE": lambda x: (sum(x) / len(x)), "COUNT": lambda x: len(x)}
-
+    result = ''
     for query, answer, predicted_agg in zip(queries, answers, aggregation_predictions_string):
         print(query)
         if predicted_agg == "NONE":
@@ -92,15 +92,13 @@ def show_answers(queries, answers, aggregation_predictions_string):
         else:
             if all([not p.match(val) == None for val in answer.split(', ')]):
                 # print("Predicted answer: " + predicted_agg + "(" + answer + ") = " + str(agg[predicted_agg](list(map(float, answer.split(','))))))
-                return "Predicted answer: " + str(agg[predicted_agg](list(map(float, answer.split(',')))))
+                result = "Predicted answer: " + str(agg[predicted_agg](list(map(float, answer.split(',')))))
             elif predicted_agg == "COUNT":
                 # print("Predicted answer: " + predicted_agg + "(" + answer + ") = " + str(agg[predicted_agg](answer.split(','))))
-                return "Predicted answer: " + str(agg[predicted_agg](answer.split(',')))
+                result = "Predicted answer: " + str(agg[predicted_agg](answer.split(',')))
             else:
-                return "Predicted answer: " + predicted_agg + " > " + answer
-
-
-
+                result = "Predicted answer: " + predicted_agg + " > " + answer
+    return result
 
 def execute_query(query, table):