Merge 0592f1b into 5b3e960

TuringLang · May 22, 2023 · 77d50dd · 77d50dd
2 parents 5b3e960 + 0592f1b
commit 77d50dd
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "MCMCDiagnosticTools"
 uuid = "be115224-59cd-429b-ad48-344e309966f0"
 authors = ["David Widmann"]
-version = "0.3.2"
+version = "0.3.3"
 
 [deps]
 AbstractFFTs = "621f4979-c628-5d54-868e-fcf4e3e8185c"
@@ -24,7 +24,7 @@ DataStructures = "0.18.3"
 Distributions = "0.25"
 MLJModelInterface = "1.6"
 SpecialFunctions = "0.8, 0.9, 0.10, 1, 2"
-StatsBase = "0.33"
+StatsBase = "0.33, 0.34"
 StatsFuns = "1"
 Tables = "1"
 julia = "1.6"

diff --git a/src/discretediag.jl b/src/discretediag.jl
@@ -150,6 +150,7 @@ function billingsley_sub(f::Array{Int,3})
 
     # transition probabilities
     mP = (mapslices(sum, f; dims=[3]) ./ mapslices(sum, mf; dims=[3]))
+    replace!(mP, NaN => 0)
     mP = reshape(mP, size(mP)[1:2])
 
     idx = findall((A .* B) .> 0)

diff --git a/test/Project.toml b/test/Project.toml
@@ -32,6 +32,6 @@ MLJLIBSVMInterface = "0.2"
 MLJModels = "0.16"
 MLJXGBoostInterface = "0.3"
 OffsetArrays = "1"
-StatsBase = "0.33"
+StatsBase = "0.33, 0.34"
 Tables = "1"
 julia = "1.6"