B
    :ÝY\*  ã               @   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                   s<   e Zd Z‡ fdd„Zdd„ Zedd„ ƒZedd„ ƒZ‡  ZS )	ÚUniformc                s4   t  |¡}t  |¡}tƒ jddtj ||¡d d S )Nr   )ÚnameÚaddress_suffixÚ
torch_dist)r   Ú	to_tensorÚsuperÚ__init__ÚtorchÚdistributionsr   )ÚselfÚlowÚhigh)Ú	__class__© ú_/global/project/projectdirs/dasrepo/etalumis/pyprob_distributed/pyprob/distributions/uniform.pyr      s    

zUniform.__init__c             C   s   d  | j| j¡S )NzUniform(lwo: {}, high: {}))Úformatr   r   )r   r   r   r   Ú__repr__   s    zUniform.__repr__c             C   s   | j jS )N)Ú_torch_distr   )r   r   r   r   r      s    zUniform.lowc             C   s   | j jS )N)r   r   )r   r   r   r   r      s    zUniform.high)	Ú__name__Ú
__module__Ú__qualname__r   r   Úpropertyr   r   Ú__classcell__r   r   )r   r   r      s   r   )r   Ú r   r   r   r   r   r   r   Ú<module>   s   