Clean up deps

Author: penelopeysm

benchmarks/Project.toml

@@ -4,12 +4,10 @@ version = "0.1.0"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
-BenchmarkTools = "6e4b80f9-dd63-53aa-95a3-0cdb28fa8baf"
 Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
 DynamicPPL = "366bfd00-2699-11ea-058f-f148b4cae6d8"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
-LogDensityProblems = "6fdf6af0-433a-55f7-b3ed-c6c6e0b8df7c"
 Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
 PrettyTables = "08abe8d2-0d0c-5749-adfa-8a2ac140af0d"
 ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
@@ -20,11 +18,9 @@ DynamicPPL = {path = "../"}
 
 [compat]
 ADTypes = "1.14.0"
-BenchmarkTools = "1.6.0"
 Distributions = "0.25.117"
 DynamicPPL = "0.37"
 ForwardDiff = "0.10.38, 1"
-LogDensityProblems = "2.1.2"
 Mooncake = "0.4"
 PrettyTables = "2.4.0"
 ReverseDiff = "1.15.3"

benchmarks/src/DynamicPPLBenchmarks.jl

@@ -1,10 +1,8 @@
 module DynamicPPLBenchmarks
 
-using DynamicPPL: VarInfo, SimpleVarInfo, VarName
-using BenchmarkTools: BenchmarkGroup, @benchmarkable
+using DynamicPPL: Model, VarInfo, SimpleVarInfo
 using DynamicPPL: DynamicPPL
 using ADTypes: ADTypes
-using LogDensityProblems: LogDensityProblems
 
 using ForwardDiff: ForwardDiff
 using Mooncake: Mooncake
@@ -55,14 +53,14 @@ function make_varinfo(model::Model, varinfo_choice::Symbol, adbackend::Symbol)
     elseif varinfo_choice == :simple_namedtuple
         SimpleVarInfo{Float64}(model(rng))
     elseif varinfo_choice == :simple_dict
-        retvals = model(rng)
-        vns = [VarName{k}() for k in keys(retvals)]
-        SimpleVarInfo{Float64}(Dict(zip(vns, values(retvals))))
+        vi = DynamicPPL.typed_varinfo(rng, model)
+        vals = DynamicPPL.values_as(vi, Dict)
+        SimpleVarInfo{Float64}(vals)
     else
         error("Unknown varinfo choice: $varinfo_choice")
     end
 
-    return DynamicPPL.link(vi, model)
+    return DynamicPPL.link!!(vi, model)
 end
 
 end # module