summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--monkey/edits.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/monkey/edits.py b/monkey/edits.py
index 0bbedef..df13940 100644
--- a/monkey/edits.py
+++ b/monkey/edits.py
@@ -54,8 +54,8 @@ def get_edits_from_trace(trace, test, id):
code = code_next
code_next = action.apply(code)
- if action.type in {'solve', 'solve_all', 'test'}:
- if action.type == 'solve' or action.type == 'solve_all':
+ if action.type in {'prolog_solve', 'test'}:
+ if action.type == 'prolog_solve':
queries.add(action.query.rstrip(' .'))
elif action.type == 'test':
correct = test(code)