Broken `rand` for `MatrixDirichlet` with wide parameter matrix

[ThijsvdLaar]

With RxInfer v3.6.0 on julia v1.10.4, when I try to sample from a MatrixDirichlet with a wide parameter matrix,

d = MatrixDirichlet(ones(3,5))
rand(d)

I would expect all columns to sum to one, but the last columns appear to be unpopulated:

3×5 Matrix{Float64}:
 0.055805  0.134324  0.0687899  8.491e-312    7.29291e-304
 0.475893  0.21299   0.36396    8.491e-312    5.0e-324
 0.468302  0.652686  0.56725    8.49104e-312  0.0

[bvdmitri]

@Nimrais or @Raphael-Tresor do you have time too look into it?

[bvdmitri]

The issue appears to happen on this line

[Raphael-Tresor]

@bvdmitri I will work on it this Thursday afternoon (26th Sept). I put it on my agenda.