Add a handle for gen_noise in stochastic.py