added check_tilde_rhs for Sampleable

TuringLang · Nov 27, 2024 · 4fc7b76 · 4fc7b76
1 parent 933e4ed
commit 4fc7b76
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/compiler.jl b/src/compiler.jl
@@ -179,6 +179,7 @@ end
 check_tilde_rhs(x::Distribution) = x
 check_tilde_rhs(x::AbstractArray{<:Distribution}) = x
 check_tilde_rhs(x::ReturnedModelWrapper) = x
+check_tilde_rhs(x::Sampleable) = Sampleable(check_tilde_rhs(x.model))
 
 """
     unwrap_right_vn(right, vn)