Skip to content

Commit 6ee75cc

Browse files
authored
Fix CI (#1992)
1 parent 316fc3b commit 6ee75cc

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

test/essential/ad.jl

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,13 +31,21 @@
3131
x = map(x->Float64(x), vi[SampleFromPrior()])
3232

3333
trackerℓ = LogDensityProblemsAD.ADgradient(TrackerAD(), ℓ)
34-
@test trackerℓ isa LogDensityProblemsAD.LogDensityProblemsADTrackerExt.TrackerGradientLogDensity
34+
if isdefined(Base, :get_extension)
35+
@test trackerℓ isa Base.get_extension(LogDensityProblemsAD, :LogDensityProblemsADTrackerExt).TrackerGradientLogDensity
36+
else
37+
@test trackerℓ isa LogDensityProblemsAD.LogDensityProblemsADTrackerExt.TrackerGradientLogDensity
38+
end
3539
@test trackerℓ.===
3640
∇E1 = LogDensityProblems.logdensity_and_gradient(trackerℓ, x)[2]
3741
@test sort(∇E1) grad_FWAD atol=1e-9
3842

3943
zygoteℓ = LogDensityProblemsAD.ADgradient(ZygoteAD(), ℓ)
40-
@test zygoteℓ isa LogDensityProblemsAD.LogDensityProblemsADZygoteExt.ZygoteGradientLogDensity
44+
if isdefined(Base, :get_extension)
45+
@test zygoteℓ isa Base.get_extension(LogDensityProblemsAD, :LogDensityProblemsADZygoteExt).ZygoteGradientLogDensity
46+
else
47+
@test zygoteℓ isa LogDensityProblemsAD.LogDensityProblemsADZygoteExt.ZygoteGradientLogDensity
48+
end
4149
@test zygoteℓ.===
4250
∇E2 = LogDensityProblems.logdensity_and_gradient(zygoteℓ, x)[2]
4351
@test sort(∇E2) grad_FWAD atol=1e-9

0 commit comments

Comments
 (0)