
    Q-j~                        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 erd dl	m
Z
 d dlZd dlmZ d d	lmZ g d
Z G d dej                         Zy)    )annotationsN)Any)overload)TYPE_CHECKING)deprecated_func)Sequence)BaseDistribution)CategoricalChoiceTypeselfnamelowhighsteplogc                  b   e Zd ZdZej
                  ddd	 	 	 	 	 	 	 	 	 	 	 d d       Z edd      ej
                  d!d              Z edd      ej
                  d!d	              Z	 edd      ej
                  d"d
              Z
ej
                  ddd	 	 	 	 	 	 	 	 	 	 	 d#d       Zeej
                  d$d              Zeej
                  d%d              Zeej
                  d&d              Zeej
                  d'd              Zeej
                  d(d              Zeej
                  	 	 	 	 	 	 d)d              Zej
                  	 	 	 	 	 	 d)d       Zej
                  d*d       Zej
                  d+d       Zej
                  d,d       Zej
                   edd      d,d              Zeej
                  d-d              Zeej
                  d.d              Zeej
                  d-d              Zeej
                  d-d              Zeej
                  d/d              Zed0d       Zy)1	BaseTrialzoBase class for trials.

    Note that this class is not supposed to be directly accessed by library users.
    NF)r   r   c                   t         NNotImplementedErrorr   s         `/Users/jameslopez/projects/TradingBot25/.venv/lib/python3.12/site-packages/optuna/trial/_base.pysuggest_floatzBaseTrial.suggest_float   s
     "!    z3.0.0z6.0.0c                    t         r   r   r   r   r   r   s       r   suggest_uniformzBaseTrial.suggest_uniform(   
     "!r   c                    t         r   r   r   s       r   suggest_loguniformzBaseTrial.suggest_loguniform-   r   r   c                    t         r   r   )r   r   r   r   qs        r   suggest_discrete_uniformz"BaseTrial.suggest_discrete_uniform2   r   r      c                   t         r   r   r   s         r   suggest_intzBaseTrial.suggest_int7   
     "!r   c                     y r    r   r   choicess      r   suggest_categoricalzBaseTrial.suggest_categorical=       ORr   c                     y r   r)   r*   s      r   r,   zBaseTrial.suggest_categoricalA   r-   r   c                     y r   r)   r*   s      r   r,   zBaseTrial.suggest_categoricalE       MPr   c                     y r   r)   r*   s      r   r,   zBaseTrial.suggest_categoricalI   s    QTr   c                     y r   r)   r*   s      r   r,   zBaseTrial.suggest_categoricalM   r0   r   c                     y r   r)   r*   s      r   r,   zBaseTrial.suggest_categoricalQ   s     !$r   c                    t         r   r   r*   s      r   r,   zBaseTrial.suggest_categoricalW   r'   r   c                    t         r   r   )r   valuer   s      r   reportzBaseTrial.report]       !!r   c                    t         r   r   r   s    r   should_prunezBaseTrial.should_prunea   r8   r   c                    t         r   r   r   keyr6   s      r   set_user_attrzBaseTrial.set_user_attre   r8   r   z3.1.0z5.0.0c                    t         r   r   r=   s      r   set_system_attrzBaseTrial.set_system_attri   r   r   c                    t         r   r   r:   s    r   paramszBaseTrial.paramsn   r   r   c                    t         r   r   r:   s    r   distributionszBaseTrial.distributionss   r   r   c                    t         r   r   r:   s    r   
user_attrszBaseTrial.user_attrsx   r   r   c                    t         r   r   r:   s    r   system_attrszBaseTrial.system_attrs}   r   r   c                    t         r   r   r:   s    r   datetime_startzBaseTrial.datetime_start   r   r   c                    t         r   r   r:   s    r   numberzBaseTrial.number   r8   r   )r   strr   floatr   rO   r   zfloat | Noner   boolreturnrO   )r   rN   r   rO   r   rO   rQ   rO   )
r   rN   r   rO   r   rO   r"   rO   rQ   rO   )r   rN   r   intr   rR   r   rR   r   rP   rQ   rR   )r   rN   r+   zSequence[None]rQ   None)r   rN   r+   zSequence[bool]rQ   rP   )r   rN   r+   zSequence[int]rQ   rR   )r   rN   r+   zSequence[float]rQ   rO   )r   rN   r+   zSequence[str]rQ   rN   )r   rN   r+   zSequence[CategoricalChoiceType]rQ   r
   )r6   rO   r   rR   rQ   rS   )rQ   rP   )r>   rN   r6   r   rQ   rS   )rQ   zdict[str, Any])rQ   zdict[str, BaseDistribution])rQ   zdatetime.datetime | None)rQ   rR   )__name__
__module____qualname____doc__abcabstractmethodr   r   r   r    r#   r&   r   r,   r7   r;   r?   rA   propertyrC   rE   rG   rI   rK   rM   r)   r   r   r   r      sZ   
 	 "	"	" 	" 		" 	" 	" 
	" 	" Wg&"  '" Wg&"  '" Wg&"  '" 	=>E""!"),"7:"EI"	" "
 R  RR  RP  PT  TP  P$$"A$	$  $ 	"""A"	" "
 	" " 	" " 	" " 	Wg&" ' " "  " "  " "  " "  " "  " " "r   r   )
__future__r   rX   typingr   r   r   optuna._deprecatedr   collections.abcr   datetimeoptuna.distributionsr	   r
   _SUGGEST_INT_POSITIONAL_ARGSABCr   r)   r   r   <module>rc      sC    " 
     . (5:  N s" s"r   