Fix bug in gradient testing

ptiede · Sep 12, 2024 · 44d2e39 · 44d2e39
1 parent ffa89f0
commit 44d2e39
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/Core/bayes.jl b/test/Core/bayes.jl
@@ -125,7 +125,7 @@ using FiniteDifferences
     tpost = asflat(post)
 
     x = prior_sample(tpost)
-    gz, = Enzyme.gradient(Enzyme.Reverse, Const(tpost), x)
+    gz = Enzyme.gradient(Enzyme.Reverse, Const(tpost), x)
     mfd = central_fdm(5,1)
     gfd, = FiniteDifferences.grad(mfd, tpost, x)
     @test gz ≈ gfd
@@ -141,7 +141,7 @@ using FiniteDifferences
     tpost = asflat(post)
     x = prior_sample(tpost)
     residual(post, Comrade.transform(tpost, x))
-    gz, = Enzyme.gradient(Enzyme.Reverse, Const(tpost), x)
+    gz = Enzyme.gradient(Enzyme.Reverse, Const(tpost), x)
     mfd = central_fdm(5,1)
     gfd, = FiniteDifferences.grad(mfd, tpost, x)
     @test gz ≈ gfd