test fixes

Author: mohamed82008

test/Turing/inference/Inference.jl

@@ -272,7 +272,6 @@ function initialize_parameters!(
     verbose::Bool=false,
     kwargs...
 )
-    islinked(spl.state.vi, spl) && invlink!(spl.state.vi, spl)
     # Get `init_theta`
     if init_theta !== nothing
         verbose && @info "Using passed-in initial variable values" init_theta

test/Turing/inference/hmc.jl

@@ -411,12 +411,12 @@ function AbstractMCMC.step!(
     spl.state.eval_num = 0
 
     Turing.DEBUG && @debug "current ϵ: $ϵ"
+    updategid!(spl.state.vi, spl)
 
     # When a Gibbs component
     if spl.selector.tag != :default
         # Transform the space
         Turing.DEBUG && @debug "X-> R..."
-        updategid!(spl.state.vi, spl)
         link!(spl.state.vi, spl, model)
         model(link(spl.state.vi), spl)
     end
@@ -517,6 +517,7 @@ function DynamicPPL.assume(
     vi,
 )
     Turing.DEBUG && _debug("assuming...")
+    updategid!(vi, vn, spl)
     r = vi[vn, dist]
     # acclogp!(vi, logpdf_with_trans(dist, r, islinked_and_trans(vi, vn)))
     # r

test/varinfo.jl

@@ -169,26 +169,26 @@ include(dir*"/test/test_utils/AllUtils.jl")
         meta = vi.metadata
 
         model(vi, SampleFromUniform())
-        @test all(x -> !istrans(vi, x), meta.vns)
+        @test all(x -> !DynamicPPL.istrans(vi, x), meta.vns)
 
         alg = HMC(0.1, 5)
         spl = Sampler(alg, model)
         v = copy(meta.vals)
-        @test islinked(link(vi))
-        @test !islinked(invlink(link(vi)))
+        @test DynamicPPL.islinked(DynamicPPL.link(vi))
+        @test !DynamicPPL.islinked(DynamicPPL.invlink(DynamicPPL.link(vi)))
         @test vi[SampleFromPrior()] == v
-        @test invlink(link(vi))[SampleFromPrior()] == v
+        @test DynamicPPL.invlink(DynamicPPL.link(vi))[SampleFromPrior()] == v
 
         vi = TypedVarInfo(vi)
         meta = vi.metadata
         alg = HMC(0.1, 5)
         spl = Sampler(alg, model)
-        @test all(x -> !istrans(vi, x), meta.s.vns)
-        @test all(x -> !istrans(vi, x), meta.m.vns)
+        @test all(x -> !DynamicPPL.istrans(vi, x), meta.s.vns)
+        @test all(x -> !DynamicPPL.istrans(vi, x), meta.m.vns)
         v_s = copy(meta.s.vals)
         v_m = copy(meta.m.vals)
         @test vi[SampleFromPrior()] == [v_s; v_m]
-        @test invlink(link(vi))[SampleFromPrior()] == [v_s; v_m]
+        @test DynamicPPL.invlink(DynamicPPL.link(vi))[SampleFromPrior()] == [v_s; v_m]
     end
     @testset "setgid!" begin
         vi = VarInfo()