BUG: handle non integer types for multinomial/dirichlet size

Closes gh-3173