Fix test

FluxML · Oct 4, 2024 · 5c6db01 · 5c6db01
1 parent 96da431
commit 5c6db01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/chainrules.jl b/test/chainrules.jl
@@ -421,7 +421,7 @@ end
     @test z2d_compiled.c.b === z2d_fallback.c.b
 
     # empty arrays => NoTangent()
-    @test z2d(ones(1, 0), ones(16, 0)) === NoTangent()
+    @test @inferred(Zygote.z2d(ones(1, 0), ones(16, 0))) === NoTangent()
 end
 
 @testset "ChainRules translation" begin