Remove deprecated Mooncake safety_on kwarg

TuringLang · Oct 24, 2024 · 6a42054 · 6a42054
1 parent dd05dc2
commit 6a42054
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/ad/utils.jl b/test/ad/utils.jl
@@ -79,8 +79,8 @@ function test_ad(f, x, broken=(); rtol=1e-6, atol=1e-6)
     end
 
     if (AD == "All" || AD == "Mooncake") && VERSION >= v"1.10"
-        rule = Mooncake.build_rrule(f, x; safety_on=false)
-        if :tapir in broken
+        rule = Mooncake.build_rrule(f, x)
+        if :Mooncake in broken
             @test_broken(
                 isapprox(
                     Mooncake.value_and_gradient!!(rule, f, x)[2][2],