Spaces:

Anni123
/

AuRoRA

Sleeping

Anni123 commited on Aug 4, 2023

Commit

5ca523c

1 Parent(s): 0a67ebb

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -22,6 +22,8 @@ def answer_cleansing_zero_shot(dataset, pred, must_choice=False):
     elif dataset == "symbolic-letter":
         pred = re.sub("\"|\'|\n|\.|\s", "", pred)
         pred = [pred]
     else:
         raise ValueError("dataset is not properly defined ...")

     elif dataset == "symbolic-letter":
         pred = re.sub("\"|\'|\n|\.|\s", "", pred)
         pred = [pred]
+    elif dataset == "UNDEFINED":
+        pred = pred
     else:
         raise ValueError("dataset is not properly defined ...")