Fix TrackedReal nextfloat and prevfloat (#111)

TuringLang · Aug 27, 2020 · 02923db · 02923db
1 parent 05c5aa8
commit 02923db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/tracker.jl b/src/tracker.jl
@@ -5,11 +5,11 @@ Tracker.dual(x::Int, p) = x
 
 Base.prevfloat(r::TrackedReal) = track(prevfloat, r)
 @grad function prevfloat(r::Real)
-    prevfloat(data(r)), Δ -> Δ
+    prevfloat(data(r)), Δ -> (Δ,)
 end
 Base.nextfloat(r::TrackedReal) = track(nextfloat, r)
 @grad function nextfloat(r::Real)
-    nextfloat(data(r)), Δ -> Δ
+    nextfloat(data(r)), Δ -> (Δ,)
 end
 
 for f = [:hcat, :vcat]