fix: remove deprecated api

LuxDL · Jul 10, 2024 · 1dac355 · 1dac355
1 parent bd18f0d
commit 1dac355
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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