o
    }o™i8  ã                   @  s‚   d dl mZ d dlZd dlmZ d dlZd dlmZ d dlmZ d dl	m
Z
 d dlmZ d dlmZ g d	¢ZG d
d„ dejƒZdS )é    )ÚannotationsN)ÚSequence)ÚAny)Úoverload)Údeprecated_func)ÚBaseDistribution)ÚCategoricalChoiceType©ÚselfÚnameÚlowÚhighÚstepÚlogc                   @  sÒ  e Zd ZdZejdddœdKdd„ƒZeddƒejdLdd„ƒƒZeddƒejdLdd„ƒƒZ	eddƒejdMdd„ƒƒZ
ejdddœdNdd„ƒZeejdOd!d"„ƒƒZeejdPd$d"„ƒƒZeejdQd&d"„ƒƒZeejdRd(d"„ƒƒZeejdSd*d"„ƒƒZeejdTd-d"„ƒƒZejdTd.d"„ƒZejdUd0d1„ƒZejdVd2d3„ƒZejdWd6d7„ƒZejed8d9ƒdWd:d;„ƒƒZeejdXd=d>„ƒƒZeejdYd@dA„ƒƒZeejdXdBdC„ƒƒZeejdXdDdE„ƒƒZeejdZdGdH„ƒƒZed[dIdJ„ƒZdS )\Ú	BaseTrialzoBase class for trials.

    Note that this class is not supposed to be directly accessed by library users.
    NF)r   r   r   Ústrr   Úfloatr   r   úfloat | Noner   ÚboolÚreturnc                C  ó   t ‚©N©ÚNotImplementedErrorr	   © r   úF/home/ubuntu/.local/lib/python3.10/site-packages/optuna/trial/_base.pyÚsuggest_float   s   
zBaseTrial.suggest_floatz3.0.0z6.0.0c                 C  r   r   r   ©r
   r   r   r   r   r   r   Úsuggest_uniform#   ó   zBaseTrial.suggest_uniformc                 C  r   r   r   r   r   r   r   Úsuggest_loguniform(   r   zBaseTrial.suggest_loguniformÚqc                 C  r   r   r   )r
   r   r   r   r!   r   r   r   Úsuggest_discrete_uniform-   r   z"BaseTrial.suggest_discrete_uniformé   Úintc                C  r   r   r   r	   r   r   r   Úsuggest_int2   ó   zBaseTrial.suggest_intÚchoicesúSequence[None]ÚNonec                 C  ó   d S r   r   ©r
   r   r'   r   r   r   Úsuggest_categorical8   ó   zBaseTrial.suggest_categoricalúSequence[bool]c                 C  r*   r   r   r+   r   r   r   r,   <   r-   úSequence[int]c                 C  r*   r   r   r+   r   r   r   r,   @   r-   úSequence[float]c                 C  r*   r   r   r+   r   r   r   r,   D   r-   úSequence[str]c                 C  r*   r   r   r+   r   r   r   r,   H   r-   úSequence[CategoricalChoiceType]r   c                 C  r*   r   r   r+   r   r   r   r,   L   r&   c                 C  r   r   r   r+   r   r   r   r,   R   r&   Úvaluec                 C  r   r   r   )r
   r3   r   r   r   r   ÚreportX   r-   zBaseTrial.reportc                 C  r   r   r   ©r
   r   r   r   Úshould_prune\   r-   zBaseTrial.should_pruneÚkeyr   c                 C  r   r   r   ©r
   r7   r3   r   r   r   Úset_user_attr`   r-   zBaseTrial.set_user_attrz3.1.0z5.0.0c                 C  r   r   r   r8   r   r   r   Úset_system_attrd   r   zBaseTrial.set_system_attrúdict[str, Any]c                 C  r   r   r   r5   r   r   r   Úparamsi   r   zBaseTrial.paramsúdict[str, BaseDistribution]c                 C  r   r   r   r5   r   r   r   Údistributionsn   r   zBaseTrial.distributionsc                 C  r   r   r   r5   r   r   r   Ú
user_attrss   r   zBaseTrial.user_attrsc                 C  r   r   r   r5   r   r   r   Úsystem_attrsx   r   zBaseTrial.system_attrsúdatetime.datetime | Nonec                 C  r   r   r   r5   r   r   r   Údatetime_start}   r   zBaseTrial.datetime_startc                 C  r   r   r   r5   r   r   r   Únumber‚   r-   zBaseTrial.number)r   r   r   r   r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   )
r   r   r   r   r   r   r!   r   r   r   )r   r   r   r$   r   r$   r   r$   r   r   r   r$   )r   r   r'   r(   r   r)   )r   r   r'   r.   r   r   )r   r   r'   r/   r   r$   )r   r   r'   r0   r   r   )r   r   r'   r1   r   r   )r   r   r'   r2   r   r   )r3   r   r   r$   r   r)   )r   r   )r7   r   r3   r   r   r)   )r   r;   )r   r=   )r   rA   )r   r$   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚabcÚabstractmethodr   r   r   r    r"   r%   r   r,   r4   r6   r9   r:   Úpropertyr<   r>   r?   r@   rB   rC   r   r   r   r   r      s€    ùÿr   )Ú
__future__r   rH   Úcollections.abcr   ÚdatetimeÚtypingr   r   Úoptuna._deprecatedr   Úoptuna.distributionsr   r   Ú_SUGGEST_INT_POSITIONAL_ARGSÚABCr   r   r   r   r   Ú<module>   s    