fix checknumerics

xukai92 · Apr 28, 2019 · aca9591 · aca9591
1 parent ef621ab
commit aca9591
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/distributions/beta.jl b/src/distributions/beta.jl
@@ -15,7 +15,7 @@ end
 function _u2logkumaraswamysample(T, u, kuma::BatchKumaraswamy)
     _one = one(T)
     _eps = eps(T)
-    return log.(_one + _eps .- exp.(log.(u) ./ (kuma.b .+ _eps))) ./ (kuma.a .+ _eps)
+    return log.(_one .- exp.(log.(u) ./ kuma.b)) ./ kuma.a
 end
 
 """

diff --git a/src/scripting.jl b/src/scripting.jl
@@ -45,7 +45,7 @@ function checknumerics(vcheck, vmonitor...; vcheckname=nothing, vmonitornames=no
     for i in eachindex(vcheck)
         check = vcheck[i]
         if isnan(check) || isinf(check)
-            ci = CartesianIndices(check)[i]
+            ci = CartesianIndices(size(vcheck))[i]
             Istr = replace(string(ci.I), r"(\s)|(\()|(\))" => "")
             if vcheckname != nothing
                 check = ("($vcheckname[$Istr], $check)")