From 855a64672af22d6c9d0800bf428f0ed0af198808 Mon Sep 17 00:00:00 2001
From: Hossein Pourbozorg <prbzrg@gmail.com>
Date: Mon, 9 Sep 2024 00:17:11 +0330
Subject: [PATCH] fix

---
 test/regression_tests.jl | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/test/regression_tests.jl b/test/regression_tests.jl
index 5393fc05..ccdc433b 100644
--- 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,