add a check for invalid output in eval

chanind · May 15, 2022 · 0e9eb88 · 0e9eb88
1 parent 81a829a
commit 0e9eb88
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/frame_semantic_transformer/data/task_samples/ArgumentsExtractionSample.py b/frame_semantic_transformer/data/task_samples/ArgumentsExtractionSample.py
@@ -101,6 +101,7 @@ def split_output_fe_spans(output: str) -> list[tuple[str, str]]:
     for span in output.split("|"):
         parts = span.strip().split("=")
         if len(parts) == 1:
-            outputs.append((parts, "XXX"))
-        outputs.append((parts[0].strip(), parts[1].strip()))
+            outputs.append((parts[0].strip(), "XXX"))
+        else:
+            outputs.append((parts[0].strip(), parts[1].strip()))
     return outputs