From 1dac355c5b5055f4fee126d2962af8cac190d46d Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Tue, 9 Jul 2024 22:55:20 -0700 Subject: [PATCH] fix: remove deprecated api --- test/others/forwarddiff_tests.jl | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/test/others/forwarddiff_tests.jl b/test/others/forwarddiff_tests.jl index 7a0b4c2a..a712bf3b 100644 --- a/test/others/forwarddiff_tests.jl +++ b/test/others/forwarddiff_tests.jl @@ -102,8 +102,9 @@ end @test_nowarn dropout(rng, x_dual, 0.5f0, Val(true), 2.0f0, :) - x_dropout = dropout(rng, x, 0.5f0, Val(true); dims=:)[1] - x_dual_dropout = ForwardDiff.value.(dropout(rng, x_dual, 0.5f0, Val(true); dims=:)[1]) + x_dropout = dropout(rng, x, 0.5f0, Val(true), 2.0f0, :)[1] + x_dual_dropout = ForwardDiff.value.(dropout( + rng, x_dual, 0.5f0, Val(true), 2.0f0, :)[1]) @test check_approx(x_dropout, x_dual_dropout) end