Added missing impl for updategid! and ThreadSafeVarInfo (#431)

TuringLang · Nov 7, 2022 · 6977dc0 · 6977dc0 · torfjelde · Nov 7, 2022
1 parent 2935493
commit 6977dc0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "DynamicPPL"
 uuid = "366bfd00-2699-11ea-058f-f148b4cae6d8"
-version = "0.21.0"
+version = "0.21.1"
 
 [deps]
 AbstractMCMC = "80f14c24-f653-4e6a-9b94-39d6b0f70001"

diff --git a/src/varinfo.jl b/src/varinfo.jl
@@ -1240,7 +1240,7 @@ end
 Set `vn`'s `gid` to `Set([spl.selector])`, if `vn` does not have a sampler selector linked
 and `vn`'s symbol is in the space of `spl`.
 """
-function updategid!(vi::VarInfo, vn::VarName, spl::Sampler)
+function updategid!(vi::VarInfoOrThreadSafeVarInfo, vn::VarName, spl::Sampler)
     if inspace(vn, getspace(spl))
         setgid!(vi, spl.selector, vn)
     end