fix more bugs

penelopeysm · penelopeysm · commit 2686fccbb104 · 2025-07-29T16:32:01.000+01:00
diff --git a/benchmarks/Project.toml b/benchmarks/Project.toml
@@ -13,6 +13,7 @@ Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
 PrettyTables = "08abe8d2-0d0c-5749-adfa-8a2ac140af0d"
 ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
 StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
+Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 
 [sources]
 DynamicPPL = {path = "../"}
@@ -27,3 +28,4 @@ Mooncake = "0.4"
 PrettyTables = "2.4.0"
 ReverseDiff = "1.15.3"
 StableRNGs = "1"
+Statistics = "1.11.1"
diff --git a/benchmarks/benchmarks.jl b/benchmarks/benchmarks.jl
@@ -3,6 +3,7 @@ using DynamicPPL.TestUtils.AD: run_ad, NoTest
 using Chairmarks: @be
 using PrettyTables: PrettyTables, ft_printf
 using StableRNGs: StableRNG
+using Statistics: median
 
 rng = StableRNG(23)
 
diff --git a/src/test_utils/ad.jl b/src/test_utils/ad.jl
@@ -308,7 +308,7 @@ function run_ad(
         )
         (median_grad, median_primal)
     else
-        nothing
+        nothing, nothing
     end
 
     return ADResult(
