B
    ɒ\                 @   s4   d dl Z ddlmZ ddlmZ G dd deZdS )    N   )Distribution   )utilc                   s0   e Zd Z fddZdd Zedd Z  ZS )Poissonc                s(   t |}t jddtj|d d S )Nr   )nameaddress_suffix
torch_dist)r   	to_tensorsuper__init__torchdistributionsr   )selfrate)	__class__ `/global/project/projectdirs/dasrepo/etalumis/pyprob_master_mar20/pyprob/distributions/poisson.pyr      s    
zPoisson.__init__c             C   s   d | jS )NzPoisson(rate: {}))formatr   )r   r   r   r   __repr__   s    zPoisson.__repr__c             C   s   | j jS )N)_torch_distmean)r   r   r   r   r      s    zPoisson.rate)__name__
__module____qualname__r   r   propertyr   __classcell__r   r   )r   r   r      s   r   )r    r   r   r   r   r   r   r   <module>   s   