Fix docs

Author: saulshanabrook

docs/reference/egglog-translation.md

@@ -311,7 +311,8 @@ path_ruleset = egraph.ruleset("path")
 # (rule ((edge x y))
 #       ((path x y)) :ruleset path)
 x, y = vars_("x y", i64)
-path_ruleset.register(rule(edge(x, y)).then(path(x, y)))
+x, y = vars_("x y", i64)
+egraph.register(rule(edge(x, y), ruleset=path_ruleset).then(path(x, y)))
 ```
 
 ### Rewrites
@@ -348,8 +349,7 @@ Rulsets can be run as well, by calling the `run` method on them:
 
 ```{code-cell} python
 # egg: (run 10 :ruleset path)
-run_report = path_ruleset.run(10)
-run_report
+egraph.run(10, ruleset=path_ruleset)
 ```
 
 After a run, you get a run report, with some timing information as well as whether things were updated.
@@ -406,19 +406,18 @@ step_egraph.register(left(i64(0)), right(i64(0)))
 x, y = vars_("x y", i64)
 
 step_left = step_egraph.ruleset("step-left")
-step_left.register(
+step_right = step_egraph.ruleset("step-right")
+step_egraph.register(
     rule(
         left(x),
         right(x),
-    ).then(left(x + 1))
-)
-
-step_right = step_egraph.ruleset("step-right")
-step_right.register(
+        ruleset=step_left
+    ).then(left(x + 1)),
     rule(
         left(x),
         right(y),
         eq(x).to(y + 1),
+        ruleset=step_right
     ).then(right(x))
 )
 

python/egglog/egraph.py

@@ -610,15 +610,15 @@ def output(self) -> None:
         raise NotImplementedError("Not imeplemented yet, because there are no examples in the egglog repo")
 
     @overload
-    def run(self, limit: int, ruleset: Optional[Ruleset] = None, /, *until: Fact) -> bindings.RunReport:
+    def run(self, limit: int, /, *until: Fact, ruleset: Optional[Ruleset] = None) -> bindings.RunReport:
         ...
 
     @overload
     def run(self, schedule: Schedule, /) -> bindings.RunReport:
         ...
 
     def run(
-        self, limit_or_schedule: int | Schedule, ruleset: Optional[Ruleset] = None, /, *until: Fact
+        self, limit_or_schedule: int | Schedule, /, *until: Fact, ruleset: Optional[Ruleset] = None
     ) -> bindings.RunReport:
         """
         Run the egraph until the given limit or until the given facts are true.