diff --git a/Project.toml b/Project.toml index eea6cdb90..56179a209 100644 --- a/Project.toml +++ b/Project.toml @@ -46,7 +46,7 @@ Requires = "1.1" PrecompileTools = "1" SpecialFunctions = "1.6, 2" Tracker = "0.2" -ZygoteRules = "0.2.1" +ZygoteRules = "0.2.4" julia = "1.6" [extensions] diff --git a/test/gradcheck.jl b/test/gradcheck.jl index bc22a2016..64d5896dd 100644 --- a/test/gradcheck.jl +++ b/test/gradcheck.jl @@ -178,7 +178,7 @@ end # Ensure that nothings work with non-numeric types. _, back = Zygote.pullback(getindex, [randn(2) for _ in 1:3], [1]) - @test back([nothing]) == (nothing, nothing) + @test back([nothing]) === nothing end @testset "view" begin