refactor: move inline marks_set_idx helper

Signed-off-by: Rudi Grinberg <[email protected]>
ocaml · Apr 16, 2024 · 0738c55 · 0738c55
1 parent a819737
commit 0738c55
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/lib/automata.ml b/lib/automata.ml
@@ -72,14 +72,12 @@ module Marks = struct
   let hash m accu =
     hash_marks_offset m.marks (hash_combine (Hashtbl.hash m.pmarks) accu)
 
-  let rec marks_set_idx idx = function
-    | (a, -1) :: rem ->
-      (a, idx) :: marks_set_idx idx rem
-    | marks ->
-      marks
-
-  let marks_set_idx marks idx =
-    { marks with marks = marks_set_idx idx marks.marks }
+  let marks_set_idx =
+    let rec marks_set_idx idx = function
+      | (a, -1) :: rem -> (a, idx) :: marks_set_idx idx rem
+      | marks -> marks
+    in
+    fun marks idx -> { marks with marks = marks_set_idx idx marks.marks }
 
   let pp_marks ch t =
     match t.marks with