Skip to content

Commit a05d2c2

Browse files
committed
pass all args even when w/o rng
1 parent 7f611fe commit a05d2c2

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/ProbProg.jl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -171,9 +171,11 @@ function sample(
171171

172172
# Remove AbstractRNG from `f`'s argument list if present, assuming that
173173
# logpdf parameters follows `(sample, args...)` convention.
174-
logpdf_args = (example_sample,)
174+
logpdf_args = nothing
175175
if !isempty(args) && args[1] isa AbstractRNG
176176
logpdf_args = (example_sample, Base.tail(args)...) # TODO: kwargs?
177+
else
178+
logpdf_args = (example_sample, args...)
177179
end
178180

179181
logpdf_mlir = invokelatest(

0 commit comments

Comments
 (0)