More test fixes

Author: penelopeysm

ext/DynamicPPLEnzymeCoreExt.jl

@@ -9,7 +9,7 @@ using EnzymeCore
     nothing
 # Likewise for get_range_and_linked.
 @inline EnzymeCore.EnzymeRules.inactive(
-    ::typeof(DynamicPPL.get_range_and_linked), args...
+    ::typeof(DynamicPPL.Experimental.get_range_and_linked), args...
 ) = nothing
 
 end

ext/DynamicPPLMooncakeExt.jl

@@ -1,10 +1,12 @@
 module DynamicPPLMooncakeExt
 
-using DynamicPPL: DynamicPPL, is_transformed, get_range_and_linked
+using DynamicPPL: DynamicPPL, is_transformed
 using Mooncake: Mooncake
 
 # This is purely an optimisation.
 Mooncake.@zero_derivative Mooncake.DefaultCtx Tuple{typeof(is_transformed),Vararg}
-Mooncake.@zero_derivative Mooncake.DefaultCtx Tuple{typeof(get_range_and_linked),Vararg}
+Mooncake.@zero_derivative Mooncake.DefaultCtx Tuple{
+    typeof(DynamicPPL.Experimental.get_range_and_linked),Vararg
+}
 
 end # module

test/fasteval.jl

@@ -5,12 +5,23 @@ using Distributions
 using DistributionsAD: filldist
 using ADTypes
 using DynamicPPL.Experimental: FastLDF
+using DynamicPPL.TestUtils.AD: run_ad, WithExpectedResult, NoTest
+using Test
+
+using ForwardDiff: ForwardDiff
+using ReverseDiff: ReverseDiff
+# Need to include this block here in case we run this test file standalone
+@static if VERSION < v"1.12"
+    using Pkg
+    Pkg.add("Mooncake")
+    using Mooncake: Mooncake
+end
 
 @testset "Automatic differentiation" begin
     # Used as the ground truth that others are compared against.
     ref_adtype = AutoForwardDiff()
 
-    test_adtypes = if MOONCAKE_SUPPORTED
+    test_adtypes = @static if VERSION < v"1.12"
         [
             AutoReverseDiff(; compile=false),
             AutoReverseDiff(; compile=true),
@@ -20,11 +31,6 @@ using DynamicPPL.Experimental: FastLDF
         [AutoReverseDiff(; compile=false), AutoReverseDiff(; compile=true)]
     end
 
-    @testset "Unsupported backends" begin
-        @model demo() = x ~ Normal()
-        @test_logs (:warn, r"not officially supported") FastLDF(demo(); adtype=AutoZygote())
-    end
-
     @testset "Correctness" begin
         @testset "$(m.f)" for m in DynamicPPL.TestUtils.DEMO_MODELS
             varinfo = VarInfo(m)