Fix test that is not broken anymore (#2349)

FluxML · Oct 26, 2023 · 4b4464a · 4b4464a
1 parent cfc6dad
commit 4b4464a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -49,7 +49,7 @@ Preferences = "1"
 ProgressLogging = "0.1"
 Reexport = "1.0"
 SpecialFunctions = "2.1.2"
-Zygote = "0.6.49"
+Zygote = "0.6.67"
 cuDNN = "1"
 julia = "1.9"
 

diff --git a/test/layers/basic.jl b/test/layers/basic.jl
@@ -421,7 +421,7 @@ end
 
   m1v = Chain([m1[1], m1[2]])  # vector of layers
   @test Zygote.hessian_dual(sum∘m1v, [1,2,3]) ≈ Zygote.hessian_dual(sum∘m1, [1,2,3])
-  @test_broken Zygote.hessian_dual(sum∘m1v, [1,2,3]) ≈ Zygote.hessian_reverse(sum∘m1v, [1,2,3])
+  @test Zygote.hessian_dual(sum∘m1v, [1,2,3]) ≈ Zygote.hessian_reverse(sum∘m1v, [1,2,3])
 
   # NNlib's softmax gradient writes in-place
   m2 = Chain(Dense(3,4,tanh), Dense(4,2), softmax)