summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTimotej Lazar <timotej.lazar@fri.uni-lj.si>2016-01-10 15:55:08 +0100
committerTimotej Lazar <timotej.lazar@fri.uni-lj.si>2016-01-10 15:55:08 +0100
commit9482cbd8644da8999051a7a6d892abb046f8d49a (patch)
tree5613242a6f85049939f38ba65c2018a25638849c
parentf740bd1d0617b35bad55b1212598de6fd98e683b (diff)
Add each edit at most once per solution
-rw-r--r--monkey/edits.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/monkey/edits.py b/monkey/edits.py
index 1863d73..f27871b 100644
--- a/monkey/edits.py
+++ b/monkey/edits.py
@@ -27,11 +27,11 @@ def get_edits_from_trace(trace, test, id):
# For each observed edit, store a list of features (with repeats) of ASTs
# where they were observed.
- edits = collections.defaultdict(list)
+ edits = collections.defaultdict(set)
def add_edit(path, start, end, tree):
if start == end:
return
- edits[(path, start, end)].append(id)
+ edits[(path, start, end)].add(id)
# Parse trace actions and ensure there is a separate action for each
# inserted/removed character.
@@ -127,7 +127,7 @@ def get_edits_from_trace(trace, test, id):
def get_edits_from_solutions(solutions, test):
# For each observed edit, store a list of features (with repeats) of ASTs
# where they were observed.
- submissions = collections.Counter()
+ submissions = collections.defaultdict(set)
queries = collections.Counter()
edits = collections.defaultdict(list)
@@ -143,7 +143,7 @@ def get_edits_from_solutions(solutions, test):
# Update submission/query counters (use normalized variables).
for code, correct in trace_submissions:
code = stringify(rename_vars(tokenize(code)))
- submissions[(code, correct)] += 1
+ submissions[(code, correct)].add(uid)
for query in trace_queries:
code = stringify(rename_vars(tokenize(query)))
queries[code] += 1