Fix Tapir tests

TuringLang · Aug 13, 2024 · 381609d · 381609d
1 parent 481b2c4
commit 381609d
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/test/ad/chainrules.jl b/test/ad/chainrules.jl
@@ -29,14 +29,17 @@ end
 
  if @isdefined Tapir
  rng = Xoshiro(123456)
- Tapir.TestUtils.test_rrule!!(
- rng, Bijectors.find_alpha, x, y, z; is_primitive=true, perf_flag=:none
+ Tapir.TestUtils.test_rule(
+ rng, Bijectors.find_alpha, x, y, z;
+ is_primitive=true, perf_flag=:none, interp=Tapir.TapirInterpreter()
  )
- Tapir.TestUtils.test_rrule!!(
- rng, Bijectors.find_alpha, x, y, 3; is_primitive=true, perf_flag=:none
+ Tapir.TestUtils.test_rule(
+ rng, Bijectors.find_alpha, x, y, 3;
+ is_primitive=true, perf_flag=:none, interp=Tapir.TapirInterpreter()
  )
- Tapir.TestUtils.test_rrule!!(
- rng, Bijectors.find_alpha, x, y, UInt32(3); is_primitive=true, perf_flag=:none
+ Tapir.TestUtils.test_rule(
+ rng, Bijectors.find_alpha, x, y, UInt32(3);
+ is_primitive=true, perf_flag=:none, interp=Tapir.TapirInterpreter()
  )
  end