From 5c6db01e6fe929173c0491435566af491565fbd2 Mon Sep 17 00:00:00 2001 From: Nicholas Bauer Date: Fri, 4 Oct 2024 18:51:23 -0400 Subject: [PATCH] Fix test --- test/chainrules.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/chainrules.jl b/test/chainrules.jl index 1835e7238..9d165d4bf 100644 --- 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