o
    8wÖiÕ  ã                   @   s8   d dl mZ d dlmZ d dlmZ G dd„ deƒZdS )é    )ÚSequence)ÚTrial)ÚCategoricalChoiceTypec                   @   sl   e Zd Zdd„ Zdedededefdd„Zddedededededefdd„Z	dede
e defdd„ZdS )Ú	FakeTrialc                 C   s   d S ©N© )Úselfr   r   úH/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/tests/utils.pyÚ__init__%   ó   zFakeTrial.__init__ÚnameÚlowÚhighÚreturnc                 C   ó   |S r   r   )r   r   r   r   r   r   r	   Úsuggest_uniform(   r   zFakeTrial.suggest_uniformé   FÚstepÚlogc                 C   r   r   r   )r   r   r   r   r   r   r   r   r	   Úsuggest_int+   r   zFakeTrial.suggest_intÚchoicesc                 C   s   |d S )Nr   r   )r   r   r   r   r   r	   Úsuggest_categorical.   s   zFakeTrial.suggest_categoricalN)r   F)Ú__name__Ú
__module__Ú__qualname__r
   ÚstrÚfloatr   ÚintÚboolr   r   r   r   r   r   r   r	   r   #   s
    $r   N)Útypingr   Úoptunar   Úoptuna.distributionsr   r   r   r   r   r	   Ú<module>   s   