minor fixes

TuringLang · Mar 15, 2020 · 510b5e5 · 510b5e5
1 parent e49f6d9
commit 510b5e5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/core/ad.jl b/src/core/ad.jl
@@ -166,7 +166,7 @@ function verifygrad(grad::AbstractVector{<:Real})
     end
 end
 
-# This seems still necessary
+# These still seem necessary
 for F in (:link, :invlink)
     @eval begin
         $F(dist::PDMatDistribution, x::Tracker.TrackedArray) = Tracker.track($F, dist, x)

diff --git a/src/inference/hmc.jl b/src/inference/hmc.jl
@@ -439,14 +439,14 @@ function assume(
     vn::VarName,
     vi::VarInfo
 )
-    Turing.DEBUG && @debug "assuming..."
+    Turing.DEBUG && _debug("assuming...")
     updategid!(vi, vn, spl)
     r = vi[vn]
     # acclogp!(vi, logpdf_with_trans(dist, r, istrans(vi, vn)))
     # r
-    Turing.DEBUG && @debug "dist = $dist"
-    Turing.DEBUG && @debug "vn = $vn"
-    Turing.DEBUG && @debug "r = $r" "typeof(r)=$(typeof(r))"
+    Turing.DEBUG && _debug("dist = $dist")
+    Turing.DEBUG && _debug("vn = $vn")
+    Turing.DEBUG && _debug("r = $r, typeof(r)=$(typeof(r))")
     return r, logpdf_with_trans(dist, r, istrans(vi, vn))
 end