Some JET fixes

src/dfa.jl

@@ -206,7 +206,7 @@ function validate_paths(
             if !eof
                 # If they are real edges but do not overlap, or there are conflicting
                 # preconditions, there is no conflict
-                overlaps(edge1, edge2) || continue
+                overlaps(edge1::Edge, edge2::Edge) || continue
             end
 
             # Now we know there is an ambiguity, so we just need to create
@@ -217,7 +217,7 @@ function validate_paths(
             final_input = if eof
                 "EOF"
             else
-                repr(Char(first(intersect(edge1.labels, edge2.labels))))
+                repr(Char(first(intersect((edge1::Edge).labels, (edge2::Edge).labels))))
             end
             error(
                 "Ambiguous NFA.\nAfter inputs $input_until_now, observing $final_input " *

src/nfa.jl

@@ -191,7 +191,7 @@ function re2nfa(re::RegExp.RE, predefined_actions::Dict{Symbol,Action}=Dict{Symb
     end
 
     nfa_start = NFANode()
-    nfa_final = rec!(nfa_start, re)
+    nfa_final::NFANode = rec!(nfa_start, re)
     return remove_dead_nodes(NFA(nfa_start, nfa_final))
 end
 

src/re.jl

@@ -55,10 +55,12 @@ end
 RE(s::AbstractString) = parse(s)
 
 function actions!(re::RE)
-    if isnothing(re.actions)
-        re.actions = Dict{Symbol, Vector{Symbol}}()
+    x = re.actions
+    if x === nothing
+        x = Dict{Symbol, Vector{Symbol}}()
+        re.actions = x
     end
-    re.actions
+    x
 end
 
 """

src/tokenizer.jl

@@ -90,7 +90,7 @@ See also: [`Tokenizer`](@ref), [`tokenize`](@ref), [`compile`](@ref)
 """
 function make_tokenizer(
     machine::TokenizerMachine;
-    tokens::Tuple{E, AbstractVector{E}}=(UInt32(1):UInt32(machine.n_tokens), UInt32(0)),
+    tokens::Tuple{E, AbstractVector{E}}=(UInt32(0), UInt32(1):UInt32(machine.n_tokens)),
     goto::Bool=true,
     version::Int=1
 ) where E
@@ -120,7 +120,7 @@ function make_tokenizer(
         )
         actions[action_name] = quote
             stop = $(vars.p)
-            token = $(nonerror_tokens[parse(Int, only(m.captures))])
+            token = $(nonerror_tokens[parse(Int, something(only(m.captures)))])
         end
     end
     return quote
@@ -209,10 +209,10 @@ function make_tokenizer(
     version::Int=1,
     unambiguous=false
 ) where E
-    (regex, _tokens) = if tokens isa Vector
-        (tokens, (UInt32(0), UInt32(1):UInt32(length(tokens))))
+    (regex, _tokens) = if tokens isa AbstractVector
+        (Vector(tokens)::Vector, (UInt32(0), UInt32(1):UInt32(length(tokens))))
     else
-        (map(last, last(tokens)), (first(tokens), map(first, last(tokens))))
+        ([last(i) for i in last(tokens)]::Vector, (first(tokens), map(first, last(tokens))))
     end
     make_tokenizer(
         compile(regex; unambiguous=unambiguous);