moved db construction to where it is used (#44)

gecBurton · Jan 30, 2021 · 162b14c · 162b14c
1 parent b65a47c
commit 162b14c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/inference_logic/algorithms.py b/inference_logic/algorithms.py
@@ -16,7 +16,6 @@
 
 
 def search(db: List, query: ImmutableDict) -> Iterator[Dict[Variable, Any]]:
-    db = [Rule(item) if not isinstance(item, Rule) else item for item in db]
     query = construct(query)
 
     i = 0
@@ -38,6 +37,8 @@ def search(db: List, query: ImmutableDict) -> Iterator[Dict[Variable, Any]]:
                 pass
         else:
             for rule in db:
+                if not isinstance(rule, Rule):
+                    rule = Rule(rule)
                 i += 1
                 rule = new_frame(rule, i)
 
@@ -54,8 +55,8 @@ def search(db: List, query: ImmutableDict) -> Iterator[Dict[Variable, Any]]:
                         ),
                     )
 
-                    for x in new_terms:
-                        stack.append((Rule(*x), new_known))
+                    for terms in new_terms:
+                        stack.append((Rule(*terms), new_known))
 
                     solutions = new_known.solutions(to_solve_for)