fix: return the result always

SciML · Sep 23, 2024 · b4d8029 · b4d8029
1 parent 229b69d
commit b4d8029
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/lib/SciMLJacobianOperators/src/SciMLJacobianOperators.jl b/lib/SciMLJacobianOperators/src/SciMLJacobianOperators.jl
@@ -150,18 +150,17 @@ function (op::JacobianOperator)(Jv, v, u, p)
     if op.mode isa VJP
         if SciMLBase.isinplace(op)
             op.vjp_op(Jv, v, u, p)
-            return
+        else
+            copyto!(Jv, op.vjp_op(v, u, p))
         end
-        copyto!(Jv, op.vjp_op(v, u, p))
-        return
     else
         if SciMLBase.isinplace(op)
             op.jvp_op(Jv, v, u, p)
-            return
+        else
+            copyto!(Jv, op.jvp_op(v, u, p))
         end
-        copyto!(Jv, op.jvp_op(v, u, p))
-        return
     end
+    return Jv
 end
 
 """