properly fixed

Author: andrewcropper

popper/gen2.py

@@ -210,7 +210,7 @@ def unsat_constraint2(self, body):
 
     def build_generalisation_constraint3(self, prog, size=None):
         rule = tuple(prog)[0]
-        for body in self.find_variants(rule):
+        for body in self.find_variants(rule, max_rule_vars=True):
             body = list(body)
             body.append((True, 'body_size', (0, len(body))))
             if size:
@@ -228,7 +228,7 @@ def build_specialisation_constraint3(self, prog, size=None):
             body.append((True, 'program_size_at_least', (size,)))
             yield frozenset(body)
 
-    def find_variants(self, rule):
+    def find_variants(self, rule, max_rule_vars=False):
         head, body = rule
         body_vars = frozenset(x for literal in body for x in literal.arguments if x >= len(head.arguments))
         if max_rule_vars:

popper/gen3.py

@@ -431,7 +431,7 @@ def build_generalisation_constraint3(self, prog, size=None):
                 return
 
             self.new_seen_rules.update(self.build_seen_rule2(rule, False))
-            for variant in self.find_variants3(rule):
+            for variant in self.find_variants3(rule, max_rule_vars=True):
                 con = []
                 con.extend(variant)
                 con.append((True, 'body_size', (0, len(body))))
@@ -452,7 +452,7 @@ def build_generalisation_constraint3(self, prog, size=None):
             bases.append(frozenset(con))
         else:
             self.new_seen_rules.update(self.build_seen_rule2(base, False))
-            for variant in self.find_variants3(base, ruleid=0):
+            for variant in self.find_variants3(base, ruleid=0, max_rule_vars=True):
                 con = []
                 con.extend(variant)
                 con.append((True, 'body_size', (0, len(base_body))))
@@ -469,7 +469,7 @@ def build_generalisation_constraint3(self, prog, size=None):
             bases.append(frozenset(con))
         else:
             self.new_seen_rules.update(self.build_seen_rule2(rec, True))
-            for variant in self.find_variants3(rec, ruleid=1):
+            for variant in self.find_variants3(rec, ruleid=1, max_rule_vars=True):
                 con = []
                 con.extend(variant)
                 con.append((True, 'body_size', (1, len(rec_body))))
@@ -501,7 +501,7 @@ def build_banish_constraint(self, prog):
 
             self.new_seen_rules.update(self.build_seen_rule2(rule, False))
 
-            for variant in self.find_variants3(rule):
+            for variant in self.find_variants3(rule, max_rule_vars=True):
                 con = []
                 con.extend(variant)
                 con.append((True, 'body_size', (0, len(body))))
@@ -521,7 +521,7 @@ def build_banish_constraint(self, prog):
             bases.append(frozenset(con))
         else:
             self.new_seen_rules.update(self.build_seen_rule2(base, False))
-            for variant in self.find_variants3(base, ruleid=0):
+            for variant in self.find_variants3(base, ruleid=0, max_rule_vars=True):
                 con = []
                 con.extend(variant)
                 con.append((True, 'body_size', (0, len(base_body))))
@@ -538,7 +538,7 @@ def build_banish_constraint(self, prog):
             bases.append(frozenset(con))
         else:
             self.new_seen_rules.update(self.build_seen_rule2(rec, True))
-            for variant in self.find_variants3(rec, ruleid=1):
+            for variant in self.find_variants3(rec, ruleid=1, max_rule_vars=True):
                 con = []
                 con.extend(variant)
                 con.append((True, 'body_size', (1, len(rec_body))))
@@ -548,7 +548,7 @@ def build_banish_constraint(self, prog):
             for r2 in recs:
                 yield r1 | r2
 
-    def find_variants3(self, rule, ruleid=0):
+    def find_variants3(self, rule, ruleid=0, max_rule_vars=False):
         head, body = rule
         head_arity = len(self.settings.head_literal.arguments)
         body_vars = frozenset(x for literal in body for x in literal.arguments if x >= head_arity)