fix test errors

Author: sunxd3

test/Project.toml

@@ -18,6 +18,7 @@ LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 LogDensityProblems = "6fdf6af0-433a-55f7-b3ed-c6c6e0b8df7c"
 MCMCChains = "c7f686f2-ff18-58e9-bc7b-31028e88f75d"
 MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09"
+Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
 OrderedCollections = "bac558e1-5e72-5ebc-8fee-abe8a469f55d"
 Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
@@ -45,6 +46,7 @@ JET = "0.9, 0.10"
 LogDensityProblems = "2"
 MCMCChains = "6.0.4, 7"
 MacroTools = "0.5.6"
+Mooncake = "0.4.137"
 OrderedCollections = "1"
 ReverseDiff = "1"
 StableRNGs = "1"

test/ext/DynamicPPLMooncakeExt.jl

@@ -3,9 +3,8 @@ using Distributions
 using Random
 using Test
 using StableRNGs
-using Mooncake: NoCache, set_to_zero!!, set_to_zero_internal!!, zero_tangent
-using DynamicPPL.TestUtils.AD: @be
-using Statistics: median
+using Mooncake: Mooncake, NoCache, set_to_zero!!, set_to_zero_internal!!, zero_tangent
+using DynamicPPL.TestUtils.AD: @be, median
 
 # Define models globally to avoid closure issues
 @model function test_model1(x)
@@ -106,6 +105,7 @@ end
         speedup = time_iddict / time_nocache
         @test speedup > 1.5  # Conservative expectation - should be ~4x
 
+        # Sanity check
         @info "Performance improvement" speedup time_iddict_μs = time_iddict / 1000 time_nocache_μs =
             time_nocache / 1000
     end
@@ -188,8 +188,5 @@ end
 
         # Global should be faster (uses NoCache)
         @test time_global < time_closure
-
-        @info "Closure handling" time_global_ms = time_global * 1000 time_closure_ms =
-            time_closure * 1000
     end
 end