Ask for which distribution to run the code in Numpy

Question

I have a code that depends on the probability distribution. By default I am using the Numpy function np.random.normal which samples from a normal distribution. So the struct