added check_tilde_rhs for Sampleable

torfjelde · torfjelde · commit 4fc7b76b566b · 2024-11-27T08:38:16.000+01:00
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)
