while using the latest version it says multinomial error(num_samples).If we try fixing this error replacing with distribution.categorical function it throws an error with importing a module nan(import nan missing).