From 2249b3f1c6c3df40c4ed0930eaecea38111044bd Mon Sep 17 00:00:00 2001 From: Kevin Ellis Date: Thu, 28 Nov 2019 20:23:53 -0500 Subject: [PATCH] fixed possible bug in regex graphs --- bin/graphs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bin/graphs.py b/bin/graphs.py index 44e8f828c..77121fa0a 100644 --- a/bin/graphs.py +++ b/bin/graphs.py @@ -148,7 +148,7 @@ def addStupidRegex(frontier, g): stupidProgram = Program.parse("(lambda (r_kleene (lambda (r_dot $0)) $0))") stupidRegex = stupidProgram.evaluate([])(pre.String("")) - if any( e.program == stupidProgram for e in frontier ): return frontier + if any( e.program == stupidProgram for e in frontier ): return frontier.normalize() lp = g.logLikelihood(frontier.task.request, stupidProgram) ll = sum(stupidRegex.match("".join(example)) for _,example in frontier.task.examples) fe = FrontierEntry(logPrior=lp, logLikelihood=ll, program=stupidProgram)