Update app/tapas.py

app/tapas.py

@@ -83,22 +83,22 @@ 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":
-            print("Predicted answer: " + answer)
-        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(','))))))
-                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(','))))
-                result = "Predicted answer: " + str(agg[predicted_agg](answer.split(',')))
-            else:
-                result = "Predicted answer: " + predicted_agg + " > " + answer
-    return result
+  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":
+          print("Predicted answer: " + answer)
+      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(','))))))
+              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(','))))
+              result = "Predicted answer: " + str(agg[predicted_agg](answer.split(',')))
+          else:
+              result = "Predicted answer: " + predicted_agg + " > " + answer
+  return result
 
 def execute_query(query, table):