Skip to content

Commit 541a2ea

Browse files
committed
uniformlogpdf out of support fix
1 parent 78e8370 commit 541a2ea

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/univariate.jl

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,14 @@ Distributions.Uniform(a::Real, b::TrackedReal) = TuringUniform{TrackedReal}(a, b
1919
Distributions.Uniform(a::TrackedReal, b::TrackedReal) = TuringUniform{TrackedReal}(a, b)
2020
Distributions.logpdf(d::Uniform, x::TrackedReal) = uniformlogpdf(d.a, d.b, x)
2121

22-
uniformlogpdf(a, b, x) = -log(b - a)
22+
function uniformlogpdf(a, b, x)
23+
c = -log(b - a)
24+
if a <= x <= b
25+
return c
26+
else
27+
return oftype(c, -Inf)
28+
end
29+
end
2330
uniformlogpdf(a::Real, b::Real, x::TrackedReal) = Tracker.track(uniformlogpdf, a, b, x)
2431
uniformlogpdf(a::TrackedReal, b::TrackedReal, x::Real) = Tracker.track(uniformlogpdf, a, b, x)
2532
uniformlogpdf(a::TrackedReal, b::TrackedReal, x::TrackedReal) = Tracker.track(uniformlogpdf, a, b, x)

0 commit comments

Comments
 (0)