fix ForwardDiffExt (look, multiple dispatch bad...)

Author: penelopeysm

ext/DynamicPPLForwardDiffExt.jl

@@ -11,7 +11,6 @@ function DynamicPPL.tweak_adtype(
     ad::ADTypes.AutoForwardDiff{chunk_size},
     ::DynamicPPL.Model,
     vi::DynamicPPL.AbstractVarInfo,
-    ::DynamicPPL.AbstractContext,
 ) where {chunk_size}
     params = vi[:]
 

test/ext/DynamicPPLForwardDiffExt.jl

@@ -14,17 +14,16 @@ using Test: @test, @testset
     @model f() = x ~ MvNormal(zeros(MODEL_SIZE), I)
     model = f()
     varinfo = VarInfo(model)
-    context = DefaultContext()
 
     @testset "Chunk size setting" for chunksize in (nothing, 0)
         base_adtype = AutoForwardDiff(; chunksize=chunksize)
-        new_adtype = DynamicPPL.tweak_adtype(base_adtype, model, varinfo, context)
+        new_adtype = DynamicPPL.tweak_adtype(base_adtype, model, varinfo)
         @test new_adtype isa AutoForwardDiff{MODEL_SIZE}
     end
 
     @testset "Tag setting" begin
         base_adtype = AutoForwardDiff()
-        new_adtype = DynamicPPL.tweak_adtype(base_adtype, model, varinfo, context)
+        new_adtype = DynamicPPL.tweak_adtype(base_adtype, model, varinfo)
         @test new_adtype.tag isa ForwardDiff.Tag{DynamicPPL.DynamicPPLTag}
     end
 end