B
    H]                 @   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 )Exponentialc                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__ c/global/project/projectdirs/dasrepo/etalumis/pyprob_master_aug5/pyprob/distributions/exponential.pyr      s    
zExponential.__init__c             C   s   d | jS )NzExponential(rate:{}))formatr   )r   r   r   r   __repr__   s    zExponential.__repr__c             C   s   | j jS )N)_torch_distmean)r   r   r   r   r      s    zExponential.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   