From b76536f4d15f605576dbab139038880e8afa2762 Mon Sep 17 00:00:00 2001 From: Timotej Lazar Date: Mon, 4 Jan 2016 12:23:55 +0100 Subject: Fix: use correct action types in monkey.edits --- monkey/edits.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'monkey') 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) -- cgit v1.2.1