Merge pull request #254 from orenht/add_visualization_flag

Add a visualize flag to egraph.saturate for further configuration

Author: saulshanabrook

python/egglog/egraph.py

@@ -1384,6 +1384,7 @@ def saturate(
         *,
         expr: Expr | None = None,
         max: int = 1000,
+        visualize: bool = True,
         **kwargs: Unpack[GraphvizKwargs],
     ) -> None:
         """
@@ -1399,18 +1400,22 @@ def to_json() -> str:
                 print(self.extract(expr), "\n")
             return self._serialize(**kwargs).to_json()
 
-        egraphs = [to_json()]
+        if visualize:
+            egraphs = [to_json()]
         i = 0
         # Always visualize, even if we encounter an error
         try:
             while (self.run(schedule or 1).updated) and i < max:
                 i += 1
-                egraphs.append(to_json())
+                if visualize:
+                    egraphs.append(to_json())
         except:
-            egraphs.append(to_json())
+            if visualize:
+                egraphs.append(to_json())
             raise
         finally:
-            VisualizerWidget(egraphs=egraphs).display_or_open()
+            if visualize:
+                VisualizerWidget(egraphs=egraphs).display_or_open()
 
     @classmethod
     def current(cls) -> EGraph: