Skip to content

Commit 8580565

Browse files
committed
fix ForwardDiffExt (look, multiple dispatch bad...)
1 parent 428b628 commit 8580565

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

ext/DynamicPPLForwardDiffExt.jl

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@ function DynamicPPL.tweak_adtype(
1111
ad::ADTypes.AutoForwardDiff{chunk_size},
1212
::DynamicPPL.Model,
1313
vi::DynamicPPL.AbstractVarInfo,
14-
::DynamicPPL.AbstractContext,
1514
) where {chunk_size}
1615
params = vi[:]
1716

test/ext/DynamicPPLForwardDiffExt.jl

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,17 +14,16 @@ using Test: @test, @testset
1414
@model f() = x ~ MvNormal(zeros(MODEL_SIZE), I)
1515
model = f()
1616
varinfo = VarInfo(model)
17-
context = DefaultContext()
1817

1918
@testset "Chunk size setting" for chunksize in (nothing, 0)
2019
base_adtype = AutoForwardDiff(; chunksize=chunksize)
21-
new_adtype = DynamicPPL.tweak_adtype(base_adtype, model, varinfo, context)
20+
new_adtype = DynamicPPL.tweak_adtype(base_adtype, model, varinfo)
2221
@test new_adtype isa AutoForwardDiff{MODEL_SIZE}
2322
end
2423

2524
@testset "Tag setting" begin
2625
base_adtype = AutoForwardDiff()
27-
new_adtype = DynamicPPL.tweak_adtype(base_adtype, model, varinfo, context)
26+
new_adtype = DynamicPPL.tweak_adtype(base_adtype, model, varinfo)
2827
@test new_adtype.tag isa ForwardDiff.Tag{DynamicPPL.DynamicPPLTag}
2928
end
3029
end

0 commit comments

Comments
 (0)