Fix scenario where the arg text is named 'value' in textinput

McGill-NLP · Oct 1, 2024 · b3f7010 · b3f7010
1 parent a6e2c63
commit b3f7010
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/weblinx/eval/metrics.py b/weblinx/eval/metrics.py
@@ -30,8 +30,14 @@ def is_applicable(self, pred, ref):
 
  def get_texts(self, pred, ref):
  if pred["intent"] == Intent.TEXT_INPUT:
- pred_text = pred["args"].get("text")
- ref_text = ref["args"].get("text")
+ if "text" not in pred["args"]:
+ pred_text = pred["args"].get("value")
+ else:
+ pred_text = pred["args"].get("text")
+ if "text" not in ref["args"]:
+ ref_text = ref["args"].get("value")
+ else:
+ ref_text = ref["args"].get("text")
  elif pred["intent"] == Intent.SAY:
  pred_text = pred["args"].get("utterance")
  ref_text = ref["args"].get("utterance")
@@ -41,7 +47,10 @@ def get_texts(self, pred, ref):
  elif pred["intent"] == Intent.CHANGE:
  pred_text = pred["args"].get("value")
  ref_text = ref["args"].get("value")
-
+ else:
+ pred_text = None
+ ref_text = None
+
  return pred_text, ref_text