NaN in tree walk

[belledon]

If there is only one observation, it's possible that one of the transition kernels returns a weighted prob of Inf, leading to a error in categorical

Also, using categorical here can be slow.