Fix relaxer bug due to datatype mismatches

potassco · Mar 4, 2024 · e535311 · e535311
1 parent 1c7ce8b
commit e535311
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/backend/src/viasp/asp/reify.py b/backend/src/viasp/asp/reify.py
@@ -708,6 +708,9 @@ def visit_ShowTerm(self, showTerm: ast.ShowTerm):  # type: ignore
 
         return new_rules
 
+    def visit_Minimize(self, minimize: ast.Minimize):  # type: ignore
+        return [minimize]
+
     def replace_anon_variables(
             self, literals: List[ast.Literal]) -> None:  # type: ignore
         """

diff --git a/backend/src/viasp/asp/utils.py b/backend/src/viasp/asp/utils.py
@@ -47,7 +47,7 @@ def merge_nodes(nodes: frozenset) -> Tuple[AST, ...]:
 
 def remove_loops(g: nx.DiGraph) -> Tuple[nx.DiGraph, FrozenSet[Tuple[AST, ...]]]:
     remove_edges: List[Tuple[AST, AST]] = []
-    where_recursion_happens: Set[AST] = set()
+    where_recursion_happens: Set[Tuple[AST]] = set()
     for edge in g.edges:
         u, v = edge
         if u == v: