Update src/reflection/reflection.jl

Co-authored-by: Paul Berg <naydex.mc+github@gmail.com>
FluxML · Jul 12, 2024 · d230579 · d230579
1 parent e5e9e57
commit d230579
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/reflection/reflection.jl b/src/reflection/reflection.jl
@@ -91,7 +91,7 @@ function invoke_tweaks!(ci::CodeInfo)
     prewalk(x) do x
       if x isa SlotNumber
         SlotNumber(x.id == 1 ? 2 : x.id+2)
-      elseif x isa Core.ReturnNode && x.val isa SlotNumber
+      elseif x isa Core.ReturnNode && isdefined(x, :val) && x.val isa SlotNumber
         Core.ReturnNode(SlotNumber(x.val.id == 1 ? 2 : x.val.id+2))
       else
         x