Skip to content

Commit b3f7010

Browse files
committed
Fix scenario where the arg text is named 'value' in textinput
1 parent a6e2c63 commit b3f7010

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

weblinx/eval/metrics.py

+12-3
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,14 @@ def is_applicable(self, pred, ref):
3030

3131
def get_texts(self, pred, ref):
3232
if pred["intent"] == Intent.TEXT_INPUT:
33-
pred_text = pred["args"].get("text")
34-
ref_text = ref["args"].get("text")
33+
if "text" not in pred["args"]:
34+
pred_text = pred["args"].get("value")
35+
else:
36+
pred_text = pred["args"].get("text")
37+
if "text" not in ref["args"]:
38+
ref_text = ref["args"].get("value")
39+
else:
40+
ref_text = ref["args"].get("text")
3541
elif pred["intent"] == Intent.SAY:
3642
pred_text = pred["args"].get("utterance")
3743
ref_text = ref["args"].get("utterance")
@@ -41,7 +47,10 @@ def get_texts(self, pred, ref):
4147
elif pred["intent"] == Intent.CHANGE:
4248
pred_text = pred["args"].get("value")
4349
ref_text = ref["args"].get("value")
44-
50+
else:
51+
pred_text = None
52+
ref_text = None
53+
4554
return pred_text, ref_text
4655

4756

0 commit comments

Comments
 (0)