Display the graphical model

google-deepmind · Jan 20, 2024 · 82364a7 · 82364a7
1 parent 1357929
commit 82364a7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/alphageometry.py b/alphageometry.py
@@ -222,6 +222,12 @@ def run_ddar(g: gh.Graph, p: pr.Problem, out_file: str) -> bool:
     return False
 
   write_solution(g, p, out_file)
+
+  gh.nm.draw(
+      g.type2nodes[gh.Point],
+      g.type2nodes[gh.Line],
+      g.type2nodes[gh.Circle],
+      g.type2nodes[gh.Segment])
   return True
 
 

diff --git a/numericals.py b/numericals.py
@@ -1207,7 +1207,7 @@ def draw(
     segments: list[gm.Segment],
     goal: Any = None,
     highlights: list[tuple[str, list[gm.Point]]] = None,
-    equals=list[tuple[Any, Any]],
+    equals: list[tuple[Any, Any]] = None,
     block: bool = True,
     save_to: str = None,
     theme: str = 'dark',