Fix test errors with LogDensityProblemsAD >= 1.2 (#1942)

Author: devmotion

test/Project.toml

@@ -44,7 +44,7 @@ DynamicPPL = "0.21.5"
 FiniteDifferences = "0.10.8, 0.11, 0.12"
 ForwardDiff = "0.10.12 - 0.10.32, 0.10"
 LogDensityProblems = "2"
-LogDensityProblemsAD = "1"
+LogDensityProblemsAD = "1.2"
 MCMCChains = "5"
 NamedArrays = "0.9.4"
 Optim = "0.22, 1.0"

test/essential/ad.jl

@@ -31,13 +31,13 @@
         x = map(x->Float64(x), vi[SampleFromPrior()])
 
         trackerℓ = LogDensityProblemsAD.ADgradient(TrackerAD(), ℓ)
-        @test trackerℓ isa LogDensityProblemsAD.TrackerGradientLogDensity
+        @test trackerℓ isa LogDensityProblemsAD.TrackerExt.TrackerGradientLogDensity
         @test trackerℓ.ℓ === ℓ
         ∇E1 = LogDensityProblems.logdensity_and_gradient(trackerℓ, x)[2]
         @test sort(∇E1) ≈ grad_FWAD atol=1e-9
 
         zygoteℓ = LogDensityProblemsAD.ADgradient(ZygoteAD(), ℓ)
-        @test zygoteℓ isa LogDensityProblemsAD.ZygoteGradientLogDensity
+        @test zygoteℓ isa LogDensityProblemsAD.ZygoteExt.ZygoteGradientLogDensity
         @test zygoteℓ.ℓ === ℓ
         ∇E2 = LogDensityProblems.logdensity_and_gradient(zygoteℓ, x)[2]
         @test sort(∇E2) ≈ grad_FWAD atol=1e-9