a
    <bY
                     @   s  d dl mZ d dlmZ d dlmZmZmZmZm	Z	m
Z
mZmZ d dlmZmZmZ d dlmZ edZedd Zeeed	d
dZeeed	ddZeeed	ddZeeed	ddZee	e	d	ddZee
e
d	ddZeeed	ddZeeed	ddZeeed	ddZeeed	ddZeeed	ddZeeed	ddZdS )    )singledispatch)import_module)BetaDistributionChiSquaredDistributionExponentialDistributionGammaDistributionLogNormalDistributionNormalDistributionParetoDistributionUniformDistribution)GeometricDistributionPoissonDistributionZetaDistribution)BinomialDistributionnumpyc                 C   s   d S )N distsizeZ
rand_stater   r   q/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/sympy/stats/sampling/sample_numpy.pydo_sample_numpy   s    r   )r   c                 C   s   |j t| jt| j |dS )N)abr   )betafloatalphar   r   r   r   _   s    r   c                 C   s   |j t| j|dS )N)Zdfr   )Z	chisquarer   kr   r   r   r   r      s    c                 C   s   |j dt| j |dS )N   r   )Zexponentialr   Zrater   r   r   r   r      s    c                 C   s   |j t| jt| j|dS Nr   )gammar   r   thetar   r   r   r   r   #   s    c                 C   s   |j t| jt| j|dS r    )Z	lognormalr   meanstdr   r   r   r   r   (   s    c                 C   s   |j t| jt| j|dS r    )normalr   r#   r$   r   r   r   r   r   -   s    c                 C   s$   t jjt| j|dd t| j S )Nr   r   r   )r   randomZparetor   r   Zxmr   r   r   r   r   2   s    c                 C   s   |j t| jt| j|dS )N)lowhighr   )uniformr   leftrightr   r   r   r   r   7   s    c                 C   s   |j t| j|dS )N)pr   )Z	geometricr   r-   r   r   r   r   r   >   s    c                 C   s   |j t| j|dS )N)Zlamr   )Zpoissonr   Zlamdar   r   r   r   r   C   s    c                 C   s   |j t| j|dS )Nr&   )Zzipfr   sr   r   r   r   r   H   s    c                 C   s   |j t| jt| j|dS )N)nr-   r   )Zbinomialintr/   r   r-   r   r   r   r   r   O   s    N)	functoolsr   Zsympy.externalr   Zsympy.stats.crv_typesr   r   r   r   r   r	   r
   r   Zsympy.stats.drv_typesr   r   r   Zsympy.stats.frv_typesr   r   r   registerr   r   r   r   r   <module>   s>   (
