Performance improvement for TA + mathprompter
Browse files
TA.py
CHANGED
|
@@ -46,8 +46,8 @@ def solve_ta(question, token):
|
|
| 46 |
return None, code
|
| 47 |
return pred, code
|
| 48 |
else:
|
| 49 |
-
res = re.findall(r"Assistant:(.*)", code)[0]
|
| 50 |
-
return res, ""
|
| 51 |
|
| 52 |
|
| 53 |
if __name__ == "__main__":
|
|
|
|
| 46 |
return None, code
|
| 47 |
return pred, code
|
| 48 |
else:
|
| 49 |
+
res = re.findall(r"Assistant:(.*)", code, re.DOTALL)[0].split("Human:")[0]
|
| 50 |
+
return res.strip(), ""
|
| 51 |
|
| 52 |
|
| 53 |
if __name__ == "__main__":
|