Simplify regenerate for distributions

probcomp · Aug 20, 2022 · af6fbfc · af6fbfc
1 parent a09d1cc
commit af6fbfc
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/distribution.jl b/src/distribution.jl
@@ -106,8 +106,7 @@ end
     (new_tr, weight, NoChange())
 end
 @inline function Gen.regenerate(tr::DistributionTrace, args::Tuple, argdiffs::Tuple, selection::AllSelection)
-    new_val = random(dist(tr), args...)
-    new_tr = DistributionTrace(new_val, args, dist(tr))
+    new_tr = simulate(dist(tr), args)
     (new_tr, 0., UnknownChange())
 end
 @inline function Gen.propose(dist::Distribution, args::Tuple)