Update OptimizationEnzymeExt.jl

wsmoses · web-flow · commit 36e1b97ddd28 · 2024-10-08T22:29:39.000-05:00
diff --git a/ext/OptimizationEnzymeExt.jl b/ext/OptimizationEnzymeExt.jl
@@ -18,7 +18,7 @@ using Core: Vararg
 end
 
 function inner_grad(mode::Mode, θ, bθ, f, p) where Mode
-    Enzyme.autodiff(Mode,
+    Enzyme.autodiff(mode,
         Const(firstapply),
         Active,
         Const(f),
