diff options
| -rw-r--r-- | monkey/edits.py | 4 |
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) |