fix

impICNF · Sep 8, 2024 · 855a646 · 855a646
1 parent b116e16
commit 855a646
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/regression_tests.jl b/test/regression_tests.jl
@@ -11,7 +11,9 @@ Test.@testset "Regression Tests" begin
         nn,
         nvars,
         naugs;
-        compute_mode = ContinuousNormalizingFlows.DIVecJacMatrixMode(ADTypes.AutoZygote()),
+        compute_mode = ContinuousNormalizingFlows.DIJacVecMatrixMode(
+            ADTypes.AutoEnzyme(; function_annotation = Enzyme.Const),
+        ),
         tspan = (0.0f0, 13.0f0),
         steer_rate = 1.0f-1,
         λ₃ = 1.0f-2,